Update app.py

app.py

@@ -1,2 +1,24 @@
 import gradio as gr
-gr.load_chat("http://96.228.37.139:4002/v1/", model="mistral-small", token="ollama").launch()
+import ollama  # Ensure ollama is available in your Space
+
+# Define chat function
+def chat_with_model(message, history):
+    try:
+        response = ollama.chat(
+            model="mistral-small",
+            messages=[{"role": "user", "content": message}]
+        )
+        return response['message']['content']
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Define Gradio Chat Interface
+chatbot = gr.ChatInterface(
+    chat_with_model,
+    title="Mistral AI Chatbot",
+    description="Chat with Mistral-Small using Ollama on Hugging Face Spaces.",
+)
+
+# Launch the Gradio app
+if __name__ == "__main__":
+    chatbot.launch()