Update app.py

app.py

@@ -104,7 +104,8 @@ def generate_answer(query: str) -> str:
         f"Question: {query}\n\n"
         f"Answer:"
     )
-    response = generator(prompt, max_length=256, do_sample=True, temperature=0.7)
+    # Use max_new_tokens to generate additional tokens beyond the prompt.
+    response = generator(prompt, max_new_tokens=100, do_sample=True, temperature=0.7)
     return response[0]["generated_text"]
 
 # ====================================================