from fastapi import FastAPI
from transformers import pipeline

#Create a new FastAPI app instance
app = FastAPI()

#Initialize the text generation pipeline
#This function will be able to generate text
#given an input
pipe = pipeline("text2text-generation",
model="google/flan-t5-small")

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

    #Return the generated text in a JSON response
    return {"output": output[0]["generated_text"]}