feat: add padding

app.py

@@ -52,7 +52,7 @@ def classify_pairs(model, tokenizer, prompts, responses):
     ]
 
     with torch.inference_mode():
-        input_ids = tokenizer(texts, truncation=True, max_length=512, return_tensors="pt").to(model.device)
+        input_ids = tokenizer(texts, padding=True, truncation=True, max_length=512, return_tensors="pt").to(model.device)
         outputs = model(**input_ids).logits
 
     return outputs