mangalnikhil21's picture
Upload 3 files
ae258ee verified
raw
history blame
No virus
596 Bytes
from fastapi import FastAPI
from transformers import pipeline
## create a new FASTAPI app instance
app=FastAPI()
# Initialize the text generation pipeline
pipe=pipeline("text2text-generation",model="google/flan-t5-small")
@app.get("/")
def home():
return {"message": "Hello World"}
# Define a function to handle the GET request at '/generate'
@app.get("/generate")
def generate(text:str):
## use the pipeline from goven input text
output = pipeline(text)
## return the generayed text in Json response
return {"output":output[0]['generated_text']}