update

app.py

@@ -540,7 +540,7 @@ class ExplainerCheckbox(Component):
                         break
 
                 opt_exp_id = max([x['id'] for x in checkbox_group_info]) + 1
-                opt_output.explainer.model = self.experiment.model
+                # opt_output.explainer.model = self.experiment.model
                 # self.experiment.manager._explainers.append(opt_output.explainer)
                 # self.experiment.manager._explainer_ids.append(opt_exp_id)
                 
@@ -548,8 +548,6 @@ class ExplainerCheckbox(Component):
                     'id': opt_exp_id,
                     'explainer': opt_output.explainer,
                 }
-                # opt_res = pickle.dumps(opt_res)
-
                 self.groups.insert_check(checkbox_group_info, self.explainer_name, opt_exp_id, opt_postprocessor_id)
                 checkbox = gr.update(label="Optimized Parameter (Optimal)", interactive=True)
                 bttn = gr.update(value="Optimized", variant="secondary")
@@ -557,12 +555,12 @@ class ExplainerCheckbox(Component):
                 return [opt_res, checkbox_group_info, checkbox, bttn]
 
             def update_exp(exp_res):
-                # exp_res = pickle.loads(exp_res)
+                exp_res['explainer'].model = self.experiment.model
                 self.experiment.manager._explainers.append(exp_res['explainer'])
                 self.experiment.manager._explainer_ids.append(exp_res['id'])
 
             self.bttn.click(optimize, inputs=[self.groups.info], outputs=[self.opt_res, self.groups.info, self.opt_check, self.bttn], queue=True, concurrency_limit=1)
-            # self.opt_res.change(update_exp, self.opt_res)
+            self.opt_res.change(update_exp, self.opt_res)
         
 
 class ExpRes(Component):