Update inference.py

inference.py

@@ -166,7 +166,7 @@ def retrain_from_feedback_csv():
         encoded = tokenizer(input_text, return_tensors="pt", padding="max_length", truncation=True, max_length=128)
 
         logits = model(encoded["input_ids"])
-        loss = F.binary_cross_entropy_with_logits(logits.squeeze(), label)
+        loss = F.binary_cross_entropy_with_logits(logits.squeeze(dim=-1), label)
         loss.backward()
         optimizer.step()
         optimizer.zero_grad()