Update app.py

app.py

@@ -1,10 +1,11 @@
+# app.py
 from fastapi import FastAPI, Request
 from transformers import pipeline
 
 app = FastAPI()
 
-# Corrigido para usar o modelo compatível com text-generation
-generator = pipeline("text-generation", model="pierreguillou/pt-small", max_length=300)
+# Modelo leve e gratuito que funciona em português
+generator = pipeline("text-generation", model="pierreguillou/tiny-random-gpt2-portuguese", max_length=200)
 
 @app.post("/gerar")
 async def gerar_historia(req: Request):
@@ -12,10 +13,10 @@ async def gerar_historia(req: Request):
     relato = body.get("relato", "")
     estilo = body.get("estilo", "")
 
-    prompt = f"Crie uma história no estilo {estilo}. Baseie-se neste relato: {relato}"
+    prompt = f"Escreva uma história no estilo {estilo} baseada neste relato: {relato}"
 
     try:
-        resultado = generator(prompt, max_new_tokens=200)[0]["generated_text"]
+        resultado = generator(prompt)[0]['generated_text']
         return {"historia": resultado}
     except Exception as e:
-        return {"erro": str(e)}
+        return {"erro": str(e)}