update

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 
 
 from sam2edit import create_demo as create_demo_edit_anything
-from sam2image import create_demo as create_demo_generate_anything
+# from sam2image import create_demo as create_demo_generate_anything
 
 
 DESCRIPTION = '# [Edit Anything](https://github.com/sail-sg/EditAnything)'

sam2edit.py

@@ -31,6 +31,7 @@ def create_demo():
     base_model_path = "stabilityai/stable-diffusion-2-inpainting"
     config_dict = OrderedDict([('SAM Pretrained(v0-1)', 'shgao/edit-anything-v0-1-1'),
                             ('LAION Pretrained(v0-3)', 'shgao/edit-anything-v0-3'),
+                            # ('LAION Pretrained(v0-3-1)', '../../edit/edit-anything-ckpt-v0-3'),
                             ])
     def obtain_generation_model(controlnet_path):
         controlnet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=torch.float16)
@@ -46,6 +47,7 @@ def create_demo():
         # pipe.to(device)
         return pipe
     global default_controlnet_path
+    global pipe
     default_controlnet_path = config_dict['LAION Pretrained(v0-3)']
     pipe = obtain_generation_model(default_controlnet_path)
 
@@ -127,10 +129,10 @@ def create_demo():
         if mask_image is None:
             mask_image = source_image["mask"]
         global default_controlnet_path
-        global pipe
         print("To Use:", config_dict[condition_model], "Current:", default_controlnet_path)
         if default_controlnet_path!=config_dict[condition_model]:
             print("Change condition model to:", config_dict[condition_model])
+            global pipe
             pipe = obtain_generation_model(config_dict[condition_model])
             default_controlnet_path = config_dict[condition_model]