Spaces:
Sleeping
Sleeping
File size: 337 Bytes
fdd2c53 1d88c77 fdd2c53 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | from fastapi import FastAPI
from transformers import pipeline
app=FastAPI()
pipe =pipeline("text2text-generation", model="google/flan-t5-small")
@app.get("/")
def home():
return {"message","Hello Userss"}
@app.get("/asking")
def generate(text:str):
output=pipe(text)
return {"output":output[0]['generated_text']}
|