File size: 339 Bytes
b7959ea 814438b 1279f5a b7959ea 0fc5ecb b7959ea |
1 2 3 4 5 6 7 8 9 10 11 12 |
from fastapi import FastAPI
from transformers import pipeline
app=FastAPI()
pipe = pipeline("text2text-generation", model="openai-community/gpt2")
@app.get("/")
def home():
return {"message":"Hello World"}
@app.get("/generate")
def generate(text:str):
output=pipe(text)
return {"output":output[0]['generated_text']}
|