Update app.py

app.py

@@ -26,16 +26,20 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map=device,
 )
 
-@app.get("/generate")
+@app.post("/generate")
 async def generate_text(request: Request):
     """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:
         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}]
     
     inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(device)