fix: return tensors

app.py

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