from fastapi import FastAPI
from transformers import pipeline

#create fastapi instance
app= FastAPI()

# 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'
    }

@app.get('/generate')

def generate(text:str):
    ##use pipeline to generate text from given input text
    output= pipe(text)
    #return generated text in json
    return {"output": output[0]['generated_text']}