app.py

from fastapi import FastAPI , Query
from pydantic import BaseModel
import joblib
import uvicorn


model = joblib.load(open('stacking_rf_xgb_model.pkl', 'rb'))
vectorizer = joblib.load(open('tfidf_vectorizer.pkl', 'rb'))


app = FastAPI()
class Textinput(BaseModel):
   message : str

@app.post("/classify")
async def classy_text(user_input : Textinput):
    text_input = user_input.message
    text_vector = vectorizer.transform([text_input])
    prediction = model.predict(text_vector)
    return {
        'message' : text_input,
        'prediction' : str(prediction)
    }

@app.get("/to_get_classify")
async def classify_get(message: str = Query(..., description="Describe your problem here")):
    text_vector = vectorizer.transform([message])
    prediction = model.predict(text_vector)
    return {
        "message": message,
        "prediction": str(prediction[0])
    }