Update app.py

app.py

@@ -1,20 +1,21 @@
 from fastapi import FastAPI, Request
 from transformers import pipeline
-import uvicorn
 
 app = FastAPI()
 
-generator = pipeline("text2text-generation", model="pierreguillou/t5-base-pt-small")
+# Corrigido para usar o modelo compatível com text-generation
+generator = pipeline("text-generation", model="pierreguillou/pt-small", max_length=300)
 
 @app.post("/gerar")
 async def gerar_historia(req: Request):
     body = await req.json()
     relato = body.get("relato", "")
     estilo = body.get("estilo", "")
+
     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)}