Update app.py

app.py

@@ -19,14 +19,16 @@ def generate_paraphrase(text, temperature):
         result = paraphrase_pipeline(
             text,
             temperature=temperature,
-            max_length=2048,  # 🚀 Set max output length (bigger than default limits)
-            min_length=500,  # 🔥 Ensures longer responses
+            max_length=2048,  # 🚀 Supports long outputs
+            max_new_tokens=1024,  # 🔥 Ensures full-length output
             top_k=50,
+            top_p=0.9,  # 🎯 More variation in responses
             do_sample=True,
-            repetition_penalty=1.2,  # 🔄 Reduces repetition
-            num_return_sequences=1,  # ✅ Ensures a single, complete paraphrase
-            num_beams=5,  # 🚀 Helps generate full-length sentences
-            early_stopping=True  # ⛔ Stops at a natural sentence end
+            repetition_penalty=1.5,  # 🛑 Stops repetitive loops
+            no_repeat_ngram_size=4,  # ⛔ Prevents repetitive phrases
+            num_return_sequences=1,  # ✅ Outputs only one high-quality paraphrase
+            num_beams=7,  # 🚀 Higher number of beams for better sentence flow
+            early_stopping=True
         )
 
         # Extract and format output properly
@@ -51,7 +53,7 @@ demo = gr.Interface(
         gr.Textbox(label="Enter text", placeholder="Type your text to paraphrase...", lines=10),  # Bigger input box
         gr.Slider(0.5, 1.5, value=1.0, step=0.1, label="Creativity (Temperature)"),
     ],
-    outputs=gr.Textbox(label="Paraphrased Text", lines=25),  # 🔥 Even more space for longer output
+    outputs=gr.Textbox(label="Paraphrased Text", lines=30),  # 🔥 More space for longer output
     title="📝 AI Paraphraser",
     description=description,
     theme="soft",  # ✅ Fix theme issue