Update model_class.py

model_class.py

@@ -228,7 +228,7 @@ class CustomGPT(
     
         return idx
 
-    def generate_response(self, input_entry, temperature=0.0, topk=None):
+    def generate_response(self, input_entry, tokenizer, temperature=0.0, topk=None):
         current_device = next(self.parameters()).device
         self.eval()
         input_text = self.format_input(input_entry)