Update app.py

app.py

@@ -5,7 +5,6 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 from fastapi import FastAPI
-from pydantic import BaseModel
 import uvicorn
 
 # === Модель ===
@@ -45,22 +44,22 @@ def respond(message, history=None):
 
     return answer
 
-# === FastAPI приложение ===
+# === Gradio интерфейс (обязательно должен быть `demo` для Hugging Face Spaces) ===
+demo = gr.ChatInterface(fn=respond, title="Иннополис Бот")
+
+# === FastAPI (опционально, если нужен API) ===
 app = FastAPI()
 
-class QuestionRequest(BaseModel):
-    question: str
+@app.get("/health")
+def health_check():
+    return {"status": "OK"}
 
 @app.post("/ask")
-def ask(request: QuestionRequest):
-    return {"answer": respond(request.question)}
-
-# === Gradio интерфейс ===
-chat_ui = gr.ChatInterface(fn=respond, title="Иннополис Бот")
-
-# === Mount Gradio UI на FastAPI ===
-app = gr.mount_gradio_app(app, chat_ui, path="/")
+async def ask(question: str):
+    return {"answer": respond(question)}
 
-# === Точка входа для запуска ===
+# === Если запускаем локально (не в Spaces) ===
 if __name__ == "__main__":
+    # Для локального теста с API
+    app = gr.mount_gradio_app(app, demo, path="/")
     uvicorn.run(app, host="0.0.0.0", port=8000)