Update model_class.py

model_class.py

@@ -175,7 +175,7 @@ class TicketGPT(
         logits = self.out_head(x) #[2,4,50257]
         return logits
 
-    def classify_review(text, model, tokenizer, max_length=None, pad_token_id=50256):
+    def predict(self, text, tokenizer, max_length=None, pad_token_id=50256):
         lookup = {
             0:"Hardware",
             1:"HR Support",
@@ -188,7 +188,7 @@ class TicketGPT(
         }
         
         current_device = next(self.parameters()).device
-        model.eval()
+        self.eval()
     
         # Prepare inputs to the model
         input_ids = tokenizer.encode(text)
@@ -203,7 +203,7 @@ class TicketGPT(
     
         # Model inference
         with torch.no_grad():
-            logits = model(input_tensor)[:, -1, :]  # Logits of the last output token
+            logits = self.forward(input_tensor)[:, -1, :]  # Logits of the last output token
         predicted_label = torch.argmax(logits, dim=-1).item()
     
         # Return the classified result