Update app.py

app.py

@@ -1,27 +1,22 @@
-import gradio as gr
-from huggingface_hub import InferenceClient
-
-client = InferenceClient(
-    model="mistralai/Mistral-7B-Instruct-v0.2"
-)
-
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
 client = InferenceClient(
     model="mistralai/Mistral-7B-Instruct-v0.2",
-    token=os.getenv("Hf_TOKEN")
+    token=os.getenv("HF_TOKEN")
 )
 
 def chat(message):
 
-    response = client.text_generation(
-        message,
-        max_new_tokens=200
+    response = client.chat_completion(
+        messages=[
+            {"role": "user", "content": message}
+        ],
+        max_tokens=200
     )
 
-    return response
+    return response.choices[0].message.content
 
 
 iface = gr.Interface(
@@ -32,5 +27,4 @@ iface = gr.Interface(
     description="Free AI assistant using Hugging Face"
 )
 
-iface.launch(server_name="0.0.0.0", server_port=7860)
-iface.launch()
+iface.launch(server_name="0.0.0.0", server_port=7860)