Upload app.py

app.py

@@ -44,7 +44,7 @@ def chat(user_input, history=[]):
     prompt = f"{formatted_history}\n\n### User:\n{user_input}\n\n### Assistant:\n"
 
     # Generate response
-    response = generator(prompt, max_length=500, do_sample=True)
+    response = generator(prompt, max_length=128, do_sample=True)
 
     # Extract only the model's generated response
     answer = response[0]["generated_text"].split("### Assistant:\n")[-1].strip()