Update app.py

app.py

@@ -111,13 +111,23 @@ Question:
 Answer in short and clear sentences.
 """
 
-    response = generator(prompt, max_new_tokens=150, do_sample=True, temperature=0.6)
+    response = generator(
+        prompt,
+        max_new_tokens=100,
+        do_sample=True,
+        temperature=0.6,
+        pad_token_id=tokenizer.eos_token_id
+    )
+
     text = response[0]["generated_text"]
 
     # Remove prompt repetition
     if prompt.strip() in text:
         text = text.split(prompt.strip())[-1].strip()
 
+    # Keep only first paragraph or sentence to avoid repetition
+    text = text.split("\n")[0].strip()
+
     return text
 
 # -----------------------------