Update app.py

app.py

@@ -172,6 +172,8 @@ def predict_batch(contents, policies):
     input_ids = tokenizer(
         input_texts,
         return_tensors="pt",
+        padding=True,
+        truncation=True
     ).input_ids.to(model.device)
 
     with torch.inference_mode():