Update Main.py

Main.py

@@ -122,7 +122,7 @@ class Main:
         input_ids, input_mask = data['input_ids'].unsqueeze(0), data['attention_mask'].unsqueeze(0)
         labels = torch.tensor([1] * input_ids.size(0), dtype=torch.long).unsqueeze(0)
         with torch.no_grad():
-            outputs = model(
+            outputs = self.model(
                 input_ids,
                 token_type_ids=None,
                 attention_mask=input_mask,