update

app.py

@@ -539,7 +539,38 @@ class ExplainerCheckbox(Component):
             self.opt_check.select(self.optimal_on_select)
 
             self.bttn = gr.Button(value="Optimize", size="sm", variant="primary")
-            self.bttn.click(self.optimize, outputs=[self.opt_check, self.bttn], queue=True, concurrency_limit=1)
+
+            @spaces.GPU
+            def optimize():
+                data_id = self.gallery.selected_index
+        
+                opt_output = self.experiment.optimize(
+                    data_ids=data_id.value,
+                    explainer_id=self.default_exp_id,
+                    metric_id=self.obj_metric,
+                    direction='maximize',
+                    sampler=SAMPLE_METHOD,
+                    n_trials=OPT_N_TRIALS,
+                )
+                
+                str_id = self.get_str_ppid(opt_output.postprocessor)
+                for pp_obj, pp_id in zip(*self.experiment.manager.get_postprocessors()):
+                    if self.get_str_ppid(pp_obj) == str_id:
+                        opt_postprocessor_id = pp_id
+                        break
+
+                opt_explainer_id = max([x['id'] for x in self.groups.info]) + 1
+                opt_output.explainer.model = self.experiment.model
+                self.experiment.manager._explainers.append(opt_output.explainer)
+                self.experiment.manager._explainer_ids.append(opt_explainer_id)
+                self.groups.insert_check(self.explainer_name, opt_explainer_id, opt_postprocessor_id)
+                self.optimal_exp_id = opt_explainer_id
+                checkbox = gr.update(label="Optimized Parameter (Optimal)", interactive=True)
+                bttn = gr.update(value="Optimized", variant="secondary")
+
+                return [checkbox, bttn]
+
+            self.bttn.click(optimize, outputs=[self.opt_check, self.bttn], queue=True, concurrency_limit=1)
         
 
 class ExpRes(Component):