google

app.py

@@ -62,9 +62,10 @@ def summarize_text(input_text, model_label, char_limit):
 
     summary_ids = model.generate(
         input_ids,
-        max_length=50,
+        max_length=20,
         min_length=10,
-        do_sample=False
+        do_sample=False,
+        no_repeat_ngram_size=2  # Restrict repetition of bigrams (2-grams)
     )
 
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
@@ -76,7 +77,7 @@ iface = gr.Interface(
     inputs=[
         gr.Textbox(lines=6, label="Enter text to summarize"),
         gr.Dropdown(choices=list(model_choices.keys()), label="Choose summarization model", value="DistilBART CNN (sshleifer/distilbart-cnn-12-6)"),
-        gr.Slider(minimum=30, maximum=200, value=80, step=1, label="Max Character Limit")
+        gr.Slider(minimum=30, maximum=200, value=65, step=1, label="Max Character Limit")
     ],
     outputs=gr.Textbox(lines=3, label="Summary (truncated to character limit)"),
     title="🔥 Fast Summarizer (GPU-Optimized)",