Update app.py

app.py

@@ -3,7 +3,7 @@ import requests
 import json
 import os
 
-API_URL = "https://api-inference.huggingface.co/models/OpenAssistant/oasst-sft-6-llama-30b-xor"
+API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-v0.3"
 API_KEY = os.getenv('API_KEY')
 
 if not API_KEY:
@@ -16,20 +16,20 @@ def generate_text(prompt):
     payload = {
         "inputs": prompt,
         "parameters": {
-            "max_new_tokens": 150,
-            "temperature": 0.1,
+            "max_length": 200,
+            "do_sample": True
         }
     }
     response = requests.post(API_URL, headers=HEADERS, json=payload)
+    
     if response.status_code == 200:
-        generated_text = response.json()[0].get("generated_text", "")
-        return generated_text
+        return response.json()
     else:
         return f"Error: {response.status_code} - {response.text}"
 
 
-description = "Enter your prompt and click Generate to see the model's response."
-title = "OpenAssistant Text Generation Interface"
+description = "Enter your prompt and click Submit to see the model's response."
+title = "Mistral Text Generation Interface"
 
 api = gr.Interface(
     fn=generate_text,