Create app.py

app.py

@@ -0,0 +1,19 @@
+from fastapi import FastAPI, Request
+from transformers import pipeline
+
+app = FastAPI()
+
+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"]
+        return {"historia": resultado}
+    except Exception as e:
+        return {"erro": str(e)}