from fastapi import FastAPI
from transformers import pipeline


app = FastAPI()
pipe = pipeline("text-classification", model="FacebookAI/roberta-large-mnli") 

# @app.get("/")
# def home():
#     return{"message":"hello rath"}

@app.get("/genrate")
def gentxt(text : str):
    output = pipe(text)
    return({"output":output[0]["generated_text"]})