Update app.py

app.py

@@ -83,8 +83,11 @@ def chat_stream(message, history):
     messages.append({"role": "user", "content": message})
 
     # Stream the response
-    response = ollama.chat(base_url='https://thanthamky-ollama-api.hf.space',model=MODEL_NAME, messages=messages, stream=True)
+    
+    client = Client(host='https://thanthamky-ollama-api.hf.space')
 
+    response = client.chat(model=MODEL_NAME, messages=messages, stream=True)
+    
     partial_message = ""
     for chunk in response:
         partial_message += chunk['message']['content']