Update model_class.py

model_class.py

@@ -174,7 +174,7 @@ class TicketGPT(
         logits = self.out_head(x) #[2,4,50257]
         return logits
 
-    def predict(self, text, tokenizer, max_length=None, pad_token_id=50256):
+    def predict(self, text, tokenizer, max_length=1024, pad_token_id=50256):
         lookup = {
             0:"Hardware",
             1:"HR Support",