Update app.py

app.py

@@ -1,14 +1,31 @@
 import gradio as gr
+import requests
+import os
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
+
+def query_mistral(prompt):
+    payload = {
+        "inputs": prompt,
+        "parameters": {"max_new_tokens": 300}
+    }
+    r = requests.post(API_URL, headers=HEADERS, json=payload)
+    return r.json()[0]["generated_text"]
 
 def chat_fn(message, history):
     if history is None:
         history = []
+
     history.append({"role": "user", "content": message})
-    history.append({"role": "assistant", "content": "AI: " + message})
+    reply = query_mistral(message)
+    history.append({"role": "assistant", "content": reply})
+
     return history
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Maind.ai 🤖")
+    gr.Markdown("## Maind.ai 🤖 (Mistral)")
     chatbot = gr.Chatbot(value=[])
     msg = gr.Textbox(placeholder="Sor...")
     msg.submit(chat_fn, [msg, chatbot], chatbot)