Update app.py

app.py

@@ -8,8 +8,14 @@ HF_TOKEN = os.environ.get("HF_TOKEN")  # Токен должен быть доб
 def generate_response(user_text):
     API_URL = "https://api-inference.huggingface.co/models/tinkoff-ai/ruDialoGPT-medium"
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    response = requests.post(API_URL, headers=headers, json={"inputs": user_text})
-    return response.json()[0]["generated_text"] if response.status_code == 200 else "Ошибка"
+    try:
+        response = requests.post(API_URL, headers=headers, json={"inputs": user_text, "wait_for_model": True})
+        if response.status_code == 200:
+            return response.json()[0]["generated_text"]
+        else:
+            return f"Ошибка: {response.status_code} - {response.text}"
+    except Exception as e:
+        return f"Ошибка при запросе к модели: {str(e)}"
 
 @app.route("/chat", methods=["POST"])
 def chat():