Update app.py

app.py

@@ -11,14 +11,11 @@ client = InferenceClient(token=HF_TOKEN, model="deepseek-ai/DeepSeek-V3.1")
 
 @app.route("/chat", methods=["POST"])
 def chat():
-    data = request.json
-    messages = data.get("messages", [])
     def gen():
-        for msg in client.chat_completion(messages):
-            if msg.choices and msg.choices[0].delta.content:
-                yield msg.choices[0].delta.content
-    return Response(stream_with_context(gen()), mimetype="text/plain")
+        for delta in client.chat_completion(request.json["messages"], stream=True):
+            yield delta
+    return Response(gen(), mimetype="text/plain")
+
 
 if __name__ == "__main__":
-    # 0.0.0.0:7860 wajib untuk HF Space
     app.run(host="0.0.0.0", port=int(os.getenv("PORT", 7860)))