Update app.py

app.py

@@ -327,6 +327,9 @@ with gr.Blocks() as demo:
                     
                     add_button  = gr.Button("Run Editing (Check Log for Completion)")
                     def run_edit_text_wrapper(
+                            mask_np_list,
+                            mask_label_list,
+                            image,
                             num_tokens,
                             guidance_scale,
                             num_sampling_steps ,
@@ -338,6 +341,9 @@ with gr.Blocks() as demo:
                             
                         run_edit_text = partial(
                             run_main,
+                            mask_np_list=mask_np_list, 
+                            mask_label_list=mask_label_list,
+                            image_gt=np.array(image),
                             load_trained=True,
                             text=True,
                             num_tokens = int(num_tokens_global.value),
@@ -353,7 +359,9 @@ with gr.Blocks() as demo:
                         return load_pil_img()
                         
                     add_button.click(run_edit_text_wrapper, 
-                        inputs = [num_tokens_global,
+                        inputs = [  mask_np_list,
+                                    mask_label_list,
+                                    image_loaded,num_tokens_global,
                                     guidance_scale,
                                     num_sampling_steps,
                                     strength ,