Update app.py

app.py

@@ -62,15 +62,21 @@ def model_chat(query: Optional[str], history: Optional[History], system: str) ->
     
     # Prepare the payload for Hugging Face Inference API
     payload = {"inputs": query, "parameters": {"max_new_tokens": 150}, "history": messages}
-    headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
+    headers = {"Authorization": f"Bearer {os.getenv('YOUR_API_TOKEN')}"}
     
     try:
         # Request generation with Hugging Face Inference API
-        response = requests.post(f"https://api-inference.huggingface.co/models/Qwen/Qwen2.5-72B-Instruct", 
+        response = requests.post(f"https://api-inference.huggingface.co/models/YourModelNameHere", 
                                  json=payload, headers=headers)
         
         if response.status_code == 200:
-            response_text = response.json().get('generated_text', '')
+            response_data = response.json()
+
+            # Handle the response as a list or dict
+            if isinstance(response_data, list):
+                response_text = response_data[0].get('generated_text', '')
+            else:
+                response_text = response_data.get('generated_text', '')
             
             # Log the chat to file
             log_history_to_file(query, response_text)