Update app.py

app.py

@@ -1,27 +1,30 @@
 import streamlit as st
-import requests
-import json
-import time
+from gradio_client import Client
 
 # Load data.json
 with open('data.json') as f:
     data = json.load(f)
 
-client_endpoint = "https://olivier-truong-mistral-super-fast.hf.space/chat"
+client_endpoint = "https://olivier-truong-mistral-super-fast.hf.space/"
 
 def ask_question(question):
     if question in data:
         return data[question]
     else:
-        response = requests.post(client_endpoint, json={"context": question})
-        if response.status_code == 200:
-            answer = response.json()['response']
-            data[question] = answer
-            with open('data.json', 'w') as f:
-                json.dump(data, f)
-            return answer
-        else:
-            return "Error: Failed to retrieve an answer."
+        client = Client(client_endpoint)
+        result = client.predict(
+            question,
+            0.9,  # Temperature
+            2000,  # Max new tokens
+            0.9,  # Top-p (nucleus sampling)
+            1.2,  # Repetition penalty
+            api_name="/chat"
+        )
+        answer = result["data"]
+        data[question] = answer
+        with open('data.json', 'w') as f:
+            json.dump(data, f)
+        return answer
 
 def typewriter(text: str, speed: int):
     tokens = text.split()