Update app.py

app.py

@@ -20,7 +20,7 @@ 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].split("User:")[0].strip()
     return assistant_message
 
 iface = gr.Interface(