Update app.py

app.py

@@ -1,29 +1,26 @@
 import gradio as gr
-from openai import OpenAI
-import os
+from huggingface_hub import InferenceClient
 
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+client = InferenceClient(
+    model="mistralai/Mistral-7B-Instruct-v0.2"
+)
 
 def chat(message):
-    try:
-        response = client.chat.completions.create(
-            model="gpt-4o-mini",
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "user", "content": message}
-            ]
-        )
-        return response.choices[0].message.content
 
-    except Exception as e:
-        return f"API error: {str(e)}"
+    response = client.text_generation(
+        message,
+        max_new_tokens=200
+    )
+
+    return response
+
 
 iface = gr.Interface(
     fn=chat,
-    inputs=gr.Textbox(label="Ask something"),
+    inputs="text",
     outputs="text",
     title="AI Assistant",
-    description="Ask anything"
+    description="Free AI assistant using Hugging Face"
 )
 
-iface.launch(server_name="0.0.0.0", server_port=7860)
+iface.launch()