Update app.py

app.py

@@ -1,22 +1,20 @@
-# app.py
 from fastapi import FastAPI, Request
 from transformers import pipeline
 
 app = FastAPI()
 
-# Modelo leve e gratuito que funciona em português
-generator = pipeline("text-generation", model="pierreguillou/tiny-random-gpt2-portuguese", max_length=200)
+# Utilizando o modelo GPT-2 pequeno para português
+generator = pipeline("text-generation", model="pierreguillou/gpt2-small-portuguese", 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"Escreva uma história no estilo {estilo} baseada neste relato: {relato}"
+    prompt = f"Crie uma história no estilo {estilo}. Baseie-se neste relato: {relato}"
 
     try:
-        resultado = generator(prompt)[0]['generated_text']
+        resultado = generator(prompt, max_new_tokens=200)[0]["generated_text"]
         return {"historia": resultado}
     except Exception as e:
-        return {"erro": str(e)}
+        return {"erro": str(e)}