from fastapi import FastAPI
from transformers import pipeline


#creating a FASTAPI app instance
app= FastAPI()

#
pipe = pipeline("text2text-generation", model="google/flan-t5-small")

@app.get("/")
def home():
    return{"message":"Hello there , How are you ?"}



@app.get("/generate")
def generate(text:str):
    ##use pipeline to generate text from given input text 
    output = pipe(text)

    #return the generate text in Json 
    return{'output':output[0]['generated_text']}