from fastapi import FastAPI
from transformers import pipeline

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

### Initialize the text generation pipeline
pipe = pipeline(
    task="text2text-generation",
    model="google-t5/t5-base"
)

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

### Define a function to handle GET request at '/generate'
@app.get("/generate")
def generate(text: str):
    output = pipe(text)
    return {"output": output[0]['generated_text']}