double pickle
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
# python image_gradio.py >> ./logs/image_gradio.log 2>&1
|
2 |
import time
|
|
|
3 |
import os
|
4 |
import gradio as gr
|
5 |
import spaces
|
@@ -519,7 +520,7 @@ class ExplainerCheckbox(Component):
|
|
519 |
|
520 |
self.bttn = gr.Button(value="Optimize", size="sm", variant="primary")
|
521 |
|
522 |
-
|
523 |
def optimize(checkbox_group_info):
|
524 |
data_id = self.gallery.selected_index
|
525 |
|
@@ -547,6 +548,8 @@ class ExplainerCheckbox(Component):
|
|
547 |
'id': opt_exp_id,
|
548 |
'explainer': opt_output.explainer,
|
549 |
}
|
|
|
|
|
550 |
self.groups.insert_check(checkbox_group_info, self.explainer_name, opt_exp_id, opt_postprocessor_id)
|
551 |
checkbox = gr.update(label="Optimized Parameter (Optimal)", interactive=True)
|
552 |
bttn = gr.update(value="Optimized", variant="secondary")
|
@@ -554,6 +557,7 @@ class ExplainerCheckbox(Component):
|
|
554 |
return [opt_res, checkbox_group_info, checkbox, bttn]
|
555 |
|
556 |
def update_exp(exp_res):
|
|
|
557 |
self.experiment.manager._explainers.append(exp_res['explainer'])
|
558 |
self.experiment.manager._explainer_ids.append(exp_res['id'])
|
559 |
|
|
|
1 |
# python image_gradio.py >> ./logs/image_gradio.log 2>&1
|
2 |
import time
|
3 |
+
import pickle
|
4 |
import os
|
5 |
import gradio as gr
|
6 |
import spaces
|
|
|
520 |
|
521 |
self.bttn = gr.Button(value="Optimize", size="sm", variant="primary")
|
522 |
|
523 |
+
@spaces.GPU
|
524 |
def optimize(checkbox_group_info):
|
525 |
data_id = self.gallery.selected_index
|
526 |
|
|
|
548 |
'id': opt_exp_id,
|
549 |
'explainer': opt_output.explainer,
|
550 |
}
|
551 |
+
opt_res = pickle.dumps(opt_res)
|
552 |
+
|
553 |
self.groups.insert_check(checkbox_group_info, self.explainer_name, opt_exp_id, opt_postprocessor_id)
|
554 |
checkbox = gr.update(label="Optimized Parameter (Optimal)", interactive=True)
|
555 |
bttn = gr.update(value="Optimized", variant="secondary")
|
|
|
557 |
return [opt_res, checkbox_group_info, checkbox, bttn]
|
558 |
|
559 |
def update_exp(exp_res):
|
560 |
+
exp_res = pickle.loads(exp_res)
|
561 |
self.experiment.manager._explainers.append(exp_res['explainer'])
|
562 |
self.experiment.manager._explainer_ids.append(exp_res['id'])
|
563 |
|