Update app.py

app.py

@@ -17,8 +17,8 @@ from omegaconf import OmegaConf
 from cldm.hack import disable_verbosity, enable_sliced_attention
 from huggingface_hub import snapshot_download
 
-snapshot_download(repo_id="xichenhku/AnyDoor_models", local_dir="./AnyDoor_models")
-snapshot_download(repo_id="xichenhku/mask_refine", local_dir="./mask_refine")
+snapshot_download(repo_id="onlineformapro/anydoor-models-ofp", local_dir="./anydoor-models-ofp")
+snapshot_download(repo_id="onlineformapro/anydoor-refine-ofp", local_dir="./anydoor-refine-ofp")
 
 cv2.setNumThreads(0)
 cv2.ocl.setUseOpenCL(False)
@@ -42,7 +42,7 @@ ddim_sampler = DDIMSampler(model)
 
 if use_interactive_seg:
     from iseg.coarse_mask_refine_util import BaselineModel
-    model_path = './mask_refine/coarse_mask_refine.pth'
+    model_path = './anydoor-refine-ofp/coarse_mask_refine.pth'
     iseg_model = BaselineModel().eval()
     weights = torch.load(model_path , map_location='cpu')['state_dict']
     iseg_model.load_state_dict(weights, strict= True)