Update app.py

app.py

@@ -1,13 +1,13 @@
 import os
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "0"
 
+from fastapi import FastAPI
+from pydantic import BaseModel
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
-import gradio as gr
-from fastapi import FastAPI
 import uvicorn
 
-# === Модель ===
+# === Загрузка модели ===
 model_id = "sberbank-ai/rugpt3medium_based_on_gpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
@@ -15,14 +15,23 @@ model = AutoModelForCausalLM.from_pretrained(model_id)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
+# Контекст для модели
 context = (
     "Университет Иннополис был основан в 2012 году. "
     "Это современный вуз в России, специализирующийся на IT и робототехнике, "
     "расположенный в городе Иннополис, Татарстан.\n"
 )
 
-def respond(message, history=None):
-    prompt = f"Прочитай текст и ответь на вопрос:\n\n{context}\n\nВопрос: {message}\nОтвет:"
+# === FastAPI приложение ===
+app = FastAPI()
+
+class QuestionRequest(BaseModel):
+    question: str
+
+@app.post("/ask")
+def generate_answer(request: QuestionRequest):
+    """Обрабатывает POST-запрос с вопросом и возвращает ответ модели."""
+    prompt = f"Прочитай текст и ответь на вопрос:\n\n{context}\n\nВопрос: {request.question}\nОтвет:"
     input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(device)
 
     with torch.no_grad():
@@ -42,24 +51,8 @@ def respond(message, history=None):
     else:
         answer = output[len(prompt):].strip()
 
-    return answer
-
-# === Gradio интерфейс (обязательно должен быть `demo` для Hugging Face Spaces) ===
-demo = gr.ChatInterface(fn=respond, title="Иннополис Бот")
-
-# === FastAPI (опционально, если нужен API) ===
-app = FastAPI()
-
-@app.get("/health")
-def health_check():
-    return {"status": "OK"}
-
-@app.post("/ask")
-async def ask(question: str):
-    return {"answer": respond(question)}
+    return {"answer": answer}
 
-# === Если запускаем локально (не в Spaces) ===
+# Точка входа для запуска сервера
 if __name__ == "__main__":
-    # Для локального теста с API
-    app = gr.mount_gradio_app(app, demo, path="/")
     uvicorn.run(app, host="0.0.0.0", port=8000)