Update app.py

app.py

@@ -34,16 +34,11 @@ class BasicAgent:
         # Run the model
         out = self.generator(
             prompt,
-            max_length=len(prompt.split()) + 20,  # short answer
-            num_return_sequences=1,
-            pad_token_id=self.generator.tokenizer.eos_token_id
+            max_new_tokens=16,       # leave room for the answer
+            return_full_text=False
         )
-        # Extract the portion after "A:"
-        text = out[0]["generated_text"]
-        # split on the last "\nA:" or just on "A:" if needed
-        answer = text.split("\nA:")[-1].strip()
-        # final cleanup: no newlines, no trailing punctuation
-        return answer.replace("\n", " ").strip()
+        answer = out[0]["generated_text"].strip()
+        return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """