Spaces:
Runtime error
Runtime error
from fastapi import FastAPI | |
from transformers import pipeline | |
#creating a FASTAPI app instance | |
app= FastAPI() | |
# | |
pipe = pipeline("text2text-generation", model="google/flan-t5-small") | |
def home(): | |
return{"message":"Hello there , How are you ?"} | |
def generate(text:str): | |
##use pipeline to generate text from given input text | |
output = pipe(text) | |
#return the generate text in Json | |
return{'output':output[0]['generated_text']} |