Update app.py

app.py

@@ -6,7 +6,9 @@ from transformers import pipeline
 generator = pipeline("text-generation", model="microsoft/phi-2", max_new_tokens=200)
 
 def chat_with_agent(user_input):
-    response = generator(user_input, do_sample=True, temperature=0.7)
+    
+    prompt = f"Q: {user_input}\nA:"
+    response = generator(prompt, do_sample=True, temperature=0.7)
     return response[0]["generated_text"]
 
 iface = gr.Interface(