Update app.py

app.py

@@ -2,9 +2,8 @@ import gradio as gr
 import os
 import requests
 
-# Получаем переменные окружения (лучше задать их в интерфейсе Hugging Face → Variables)
 HF_TOKEN = os.getenv("HF_API_TOKEN")
-MODEL_NAME = os.getenv("MODEL_NAME", "google/flan-t5-small")
+MODEL_NAME = os.getenv("MODEL_NAME", "google/flan-t5-small")  # Обязательно существующая модель
 
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
@@ -13,27 +12,22 @@ def chat_fn(message, history):
     try:
         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}"
 
         try:
             result = response.json()
         except Exception:
-            return f"❌ Неверный ответ от API (не JSON):\n{response.text}"
+            return f"❌ Неверный JSON:\n{response.text}"
 
-        # Извлечение текста из ответа
         if isinstance(result, list) and "generated_text" in result[0]:
             return result[0]["generated_text"]
-        elif isinstance(result, dict) and "generated_text" in result:
-            return result["generated_text"]
         else:
             return "❌ Ответ не распознан"
     except Exception as e:
-        return f"❌ Общая ошибка: {str(e)}"
+        return f"❌ Ошибка: {str(e)}"
 
 gr.ChatInterface(
     fn=chat_fn,