Update app.py

app.py

@@ -33,8 +33,13 @@ generator = pipeline(
 )
 
 def generate_reply(prompt):
-    full_prompt = spartan_intro + "\nUser: " + prompt + "\nSpartan Mentor:"
-    result = generator(full_prompt)[0]["generated_text"]
+    full_prompt = (
+        spartan_intro.strip() + "\n\n"
+        "The Spartan Mentor is ready to guide you.\n"
+        "User: " + prompt.strip() + "\n"
+        "Spartan Mentor:"
+    )
+    result = generator(full_prompt, max_length=300, num_return_sequences=1)[0]["generated_text"]
     return result.split("Spartan Mentor:")[-1].strip()
 
 iface = gr.Interface(