Code Change

app.py

@@ -49,6 +49,10 @@ def chat():
 
         raw_output = data[0]['generated_text'] if isinstance(data, list) else data.get('generated_text', "No response")
         reply = raw_output.split("Bot:")[-1].strip()
+
+        # Remove trailing "User" if present
+        if reply.endswith("User"):
+            reply = reply[:-4].strip()
         return jsonify({"reply": reply})
 
     except Exception as e: