Spaces:
Sleeping
Sleeping
| from fastapi import FastAPI | |
| from transformers import pipeline | |
| ##create a new fastapi app instance | |
| app=FastAPI() | |
| pipe = pipeline("text2text-generation", model="Yihui/t5-small-text-summary-generation") | |
| def home(): | |
| return {"message":"hw"} | |
| def generate(text:str): | |
| output=pipe(text) | |
| return {"output":output[0]['generated_text']} | |