Update app.py

app.py

@@ -43,7 +43,7 @@ def pil_to_binary_mask(pil_image, threshold=0):
 example_path = os.path.join(os.path.dirname(__file__), 'example')
 
 unet = UNet2DConditionModel.from_pretrained(
-    base_path,
+    base_path = 'yisol/IDM-VTON'
     subfolder="unet",
     torch_dtype=torch.float16,
 )