main.py

from fastapi import FastAPI, Query
from model import model  # Импортируйте вашу модель

app = FastAPI()

@app.get("/", response_model=int)
async def predict(text: str = Query(...)):
    """

    Эндпоинт для предсказания. Принимает текстовый параметр в запросе.

    

    :param text: текст для предсказания

    :return: предсказание модели

    """
    # Получаем предсказание от модели
    output = model.predict(text)
    
    # Возвращаем результат
    return output

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="127.0.0.1", port=8000)