update

app.py

@@ -484,12 +484,12 @@ class ExplainerCheckbox(Component):
         metric_info = self.experiment.manager.get_metrics()
         idx = [metric.__class__.__name__ for metric in metric_info[0]].index(metric_name)
         return metric_info[1][idx]
+    
+    def get_str_ppid(self, pp_obj):
+        return pp_obj.pooling_fn.__class__.__name__ + pp_obj.normalization_fn.__class__.__name__
 
-    # @spaces.GPU
+    @spaces.GPU
     def optimize(self):
-        # if self.explainer_name in ["Lime", "KernelShap", "IntegratedGradients"]:
-        #     gr.Info("Lime, KernelShap and IntegratedGradients currently do not support hyperparameter optimization.")
-        #     return [gr.update()] * 2
         data_id = self.gallery.selected_index
         
         opt_output = self.experiment.optimize(
@@ -501,13 +501,9 @@ class ExplainerCheckbox(Component):
             n_trials=OPT_N_TRIALS,
         )
         
-
-        def get_str_ppid(pp_obj):
-            return pp_obj.pooling_fn.__class__.__name__ + pp_obj.normalization_fn.__class__.__name__
-        
-        str_id = get_str_ppid(opt_output.postprocessor)
+        str_id = self.get_str_ppid(opt_output.postprocessor)
         for pp_obj, pp_id in zip(*self.experiment.manager.get_postprocessors()):
-            if get_str_ppid(pp_obj) == str_id:
+            if self.get_str_ppid(pp_obj) == str_id:
                 opt_postprocessor_id = pp_id
                 break