Spaces:
Runtime error
Runtime error
from fastapi import FastAPI | |
from transformers import pipeline | |
#create a new FASTAPI App instances | |
app=FastAPI() | |
##create a new pipeline instances | |
pipe = pipeline("text2text-generation", model="google/flan-t5-small") | |
def home(): | |
return{"message":"Hello World !"} | |
#Define a function o handle the GET request at '/generate' | |
def generate(text:str): | |
###use the pipeline to generate text from given input text | |
output=pipe(text) | |
##return the generate text in Json response | |
return{"output":output[0][' ']} | |