Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import os
 import requests
+from dotenv import load_dotenv
+
+load_dotenv()
 
 HF_TOKEN = os.getenv("HF_API_TOKEN")
 MODEL_NAME = os.getenv("MODEL_NAME", "google/flan-t5-base")
@@ -10,13 +13,12 @@ HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
 
 def chat_fn(message, history):
     try:
-        payload = {"inputs": message}
+        prompt = f"Answer the question: {message}"
+        payload = {"inputs": prompt}
         response = requests.post(API_URL, headers=HEADERS, json=payload)
-        if response.status_code != 200:
-            return f"❌ Ошибка API: {response.status_code}\n{response.text}"
         result = response.json()
+        print(result)
 
-        # Обработка разных форматов ответа
         if isinstance(result, list) and "generated_text" in result[0]:
             return result[0]["generated_text"]
         elif isinstance(result, dict) and "generated_text" in result: