Update app.py

app.py

@@ -62,6 +62,8 @@ def process_depth(
     path_input,
     processing_res=default_image_processing_res,
 ):
+    print('line 65', path_input)
+    
     name_base, name_ext = os.path.splitext(os.path.basename(path_input))
     print(f"Processing image {name_base}{name_ext}")
 
@@ -518,7 +520,7 @@ def main():
     dtype = torch.float16
     
     vae = AutoencoderKL.from_pretrained("guangkaixu/GenPercept", subfolder='vae').to(dtype)
-    unet_depth_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_depth_v1").to(dtype)
+    unet_depth_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept-depth', subfolder="unet").to(dtype)
     unet_normal_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_normal_v1").to(dtype)
     unet_dis_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_dis_v1").to(dtype)