Update model_class.py

model_class.py

@@ -1,7 +1,6 @@
 import torch
 import torch.nn as nn
 from transformers import PreTrainedModel, PretrainedConfig
-# from huggingface_hub import PyTorchModelHubMixin
 
 
 class MultiheadAttention(nn.Module):
@@ -203,7 +202,7 @@ class TicketGPT(
     
         # Model inference
         with torch.no_grad():
-            logits = self.forward(input_tensor)[:, -1, :]  # Logits of the last output token
+            logits = self(input_tensor)[:, -1, :]  # Logits of the last output token
         predicted_label = torch.argmax(logits, dim=-1).item()
     
         # Return the classified result