from fastapi import FastAPI
from transformers import pipeline

## Create a FastAPI app
app = FastAPI()

#intializatiom of the pipeline
pipe=pipeline('text2text-generation',model='google/flan-t5-small')
@app.get("/")
def home():
    return {"message": "Hello World"}
#define to function to handle the get request
@app.get("/generate")
def generate(text:str):
    ## use the pipeline to generate the text from given input text
    output=pipe(text)
    ## return the generated text
    return {"output":output[0]['generated_text']}