Spaces:
Sleeping
Sleeping
File size: 641 Bytes
6352a01 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from fastapi import FastAPI
from transformers import pipeline
app = FastAPI()
pipe = pipeline(task="text-generation", model="gpt2-large", framework="pt")
@app.get("/")
def root():
"""
Returns home page.
"""
return {"message": "Hello Ismael"}
@app.get("/generate")
def generate(text: str):
"""
Using the text-generation pipeline from `transformers`, generate text
from the given input text. The model used is `openai-community/gpt2-large`, which
can be found [here](<https://huggingface.co/openai-community/gpt2-large>).
"""
output = pipe(text)
return {"output": output[0]["generated_text"]}
|