double pickle

app.py

@@ -1,5 +1,6 @@
 # python image_gradio.py >> ./logs/image_gradio.log 2>&1
 import time
+import pickle
 import os
 import gradio as gr
 import spaces
@@ -519,7 +520,7 @@ class ExplainerCheckbox(Component):
 
             self.bttn = gr.Button(value="Optimize", size="sm", variant="primary")
 
-            # @spaces.GPU
+            @spaces.GPU
             def optimize(checkbox_group_info):
                 data_id = self.gallery.selected_index
         
@@ -547,6 +548,8 @@ 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")
@@ -554,6 +557,7 @@ class ExplainerCheckbox(Component):
                 return [opt_res, checkbox_group_info, checkbox, bttn]
 
             def update_exp(exp_res):
+                exp_res = pickle.loads(exp_res)
                 self.experiment.manager._explainers.append(exp_res['explainer'])
                 self.experiment.manager._explainer_ids.append(exp_res['id'])