Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
3 |
|
4 |
app = FastAPI()
|
5 |
|
6 |
-
generator = pipeline("text-generation", model="
|
7 |
|
8 |
@app.post("/gerar")
|
9 |
async def gerar_historia(req: Request):
|
|
|
3 |
|
4 |
app = FastAPI()
|
5 |
|
6 |
+
generator = pipeline("text-generation", model="pierreguillou/pt-gpt2-medium", max_length=300)
|
7 |
|
8 |
@app.post("/gerar")
|
9 |
async def gerar_historia(req: Request):
|