Update app.py

app.py

@@ -1,9 +1,10 @@
 from fastapi import FastAPI, Request
 from transformers import pipeline
+import uvicorn
 
 app = FastAPI()
 
-generator = pipeline("text-generation", model="pierreguillou/pt-gpt2-medium", max_length=300)
+generator = pipeline("text2text-generation", model="pierreguillou/t5-base-pt-small")
 
 @app.post("/gerar")
 async def gerar_historia(req: Request):
@@ -13,7 +14,7 @@ async def gerar_historia(req: Request):
     prompt = f"Crie uma história no estilo {estilo}. Baseie-se neste relato: {relato}"
 
     try:
-        resultado = generator(prompt, max_new_tokens=200)[0]["generated_text"]
+        resultado = generator(prompt, max_new_tokens=200)[0]['generated_text']
         return {"historia": resultado}
     except Exception as e:
         return {"erro": str(e)}