Update app.py

app.py

@@ -13,6 +13,7 @@ def format_chat_prompt(message, instruction):
     return prompt
 
 def query(payload):
+    payload["parameters"] = {"stop_sequences": ["\nUser:", ""]}
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
@@ -20,7 +21,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].split("User:")[0].strip()
+    assistant_message = generated_text.split("Assistant:")[-1].strip()
     return assistant_message
 
 iface = gr.Interface(