Update app.py

app.py

@@ -39,7 +39,7 @@ Does the human question violate the above policy? Your answer must start with 'Y
 
 def verificar_politica(message, policy):
     prompt = generar_prompt(message, policy)
-    inputs = tokenizer(prompt, return_tensors="pt")
+    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     
     with torch.no_grad():
         outputs = model(**inputs)