Update app.py

app.py

@@ -21,7 +21,8 @@ def respond(message, instruction="A conversation between a user and an AI assist
     prompt = format_chat_prompt(message, instruction)
     response = query({"inputs": prompt})
     generated_text = response[0]['generated_text']
-    assistant_message = generated_text.split("Assistant:")[-1].strip()
+    assistant_message = generated_text.split("Assistant:")[-1]
+    assistant_message = assistant_message.split("User:")[0].strip()  # Only keep the text before the first "User:"
     return assistant_message
 
 iface = gr.Interface(