Update app.py

app.py

@@ -249,6 +249,9 @@ with gr.Blocks() as demo:
                 
                 add_button  = gr.Button("Run optimization")
                 def run_optimization_wrapper (
+                        mask_list,
+                        mask_label_list,
+                        image
                         opt_flag,                     
                         num_tokens,
                         embedding_learning_rate , 
@@ -260,7 +263,10 @@ with gr.Blocks() as demo:
                 ):
                     if 1:
                         run_optimization = partial(
-                            run_main,                 
+                            run_main,  
+                            mask_list=mask_list, 
+                            mask_label_list=mask_label_list,
+                            image_gt=image,
                             num_tokens=int(num_tokens),
                             embedding_learning_rate = float(embedding_learning_rate), 
                             max_emb_train_steps = int(max_emb_train_steps), 
@@ -281,6 +287,9 @@ with gr.Blocks() as demo:
                 
                 add_button.click(run_optimization_wrapper, 
                         inputs = [
+                            mask_np_list，
+                            mask_label_list,
+                            image_loaded,
                             opt_flag,
                             num_tokens,
                             embedding_learning_rate ,