Update chess_board.py

chess_board.py

@@ -17,8 +17,11 @@ class Game:
         self.counter = 0
         self.model_id = 'kaggle://valentinbaltazar/gemma-chess/keras/gemma_2b_en_chess'
         self.sampler = keras_nlp.samplers.TopKSampler(k=50, temperature=0.7)
-        self.model = keras_nlp.models.GemmaCausalLM.from_preset(self.model_id).compile(sampler=self.sampler)
-        
+        self.model = keras_nlp.models.GemmaCausalLM.from_preset(self.model_id)
+        self.compile()
+
+    def compile_model(self):
+        self.model.compile(sampler=self.sampler)
     
     def call_gemma(self):
         template = "Instruction:\n{instruction}\n\nResponse:\n{response}"