Update app.py

app.py

@@ -27,20 +27,16 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 
 @app.post("/generate")
-async def generate_text(request: Request):
+async def generate_text(message: str, token: str):
     """Gera um texto com base na entrada fornecida."""
-        
-    data = await request.json()
-    user_message = data.get("message")
-    token = data.get('token','null')
     
-    if not user_message:
+    if not message:
         raise HTTPException(status_code=400, detail="O campo 'message' é obrigatório.")
 
     if token != EXPECTED_TOKEN:
         raise HTTPException(status_code=401, detail="Token inválido")
         
-    messages = [{'role': 'user', 'content': user_message}]
+    messages = [{'role': 'user', 'content': message}]
     
     inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(device)