from fastapi import FastAPI
from transformers import pipeline


# Creating a new FASTAPI app instance  
app = FastAPI()

# Initialize the text generation pipeline 
pipe= pipeline()
# Use a pipeline as a high-level helper 
pipe = pipeline("text2text-generation", model="google/flan-t5-small")

@app.get("/")
def home():
    return{"message":"Hello World"}

# Defining the function to handle the get req.
@ app.get("/generate")
def generate_text(inp:str):
    #Pipeline to generate text from given input text 
    output = pipe(text)
    #Return the generated text in Json response 
    return{"output":output[0]['generated_text']}