Update app.py

app.py

@@ -1,29 +1,22 @@
 import gradio as gr
-import requests
 import os
+from huggingface_hub import InferenceClient
 
 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)
-    data = r.json()
-
-    # Eğer model yükleniyorsa
-    if isinstance(data, dict) and "error" in data:
-        return f"Model yükleniyor veya hata oluştu:\n{data['error']}"
+client = InferenceClient(
+    provider="hf-inference",
+    api_key=HF_TOKEN,
+)
 
-    # Normal cevap
-    if isinstance(data, list):
-        return data[0].get("generated_text", "Cevap alınamadı.")
-
-    return "Beklenmeyen API cevabı."
+def query_mistral(prompt):
+    response = client.text_generation(
+        model="mistralai/Mistral-7B-Instruct-v0.2",
+        prompt=prompt,
+        max_new_tokens=300,
+        temperature=0.7,
+    )
+    return response
 
 def chat_fn(message, history):
     if history is None: