Update app.py

app.py

@@ -38,7 +38,6 @@ pipe_depth = StableDiffusionControlNetPipeline.from_pretrained(
 # Load a quick, highly accurate MiDaS model pipeline via torch hub
 midas = torch.hub.load("intel-isl/MiDaS", "MiDaS_small").to(device).eval()
 midas_transforms = torch.hub.load("intel-isl/MiDaS", "transforms")
-transform_depth = midas_transforms.flag_transform if hasattr(midas_transforms, 'flag_transform') else midas_transforms.dpt_transform
 
 
 @spaces.GPU
@@ -121,6 +120,7 @@ def extract_depth_map(image_path):
     """
     Utilizes a lightweight MiDaS pipeline to extract depth estimation layouts.
     """
+    transform_depth = midas_transforms.flag_transform if hasattr(midas_transforms, 'flag_transform') else midas_transforms.dpt_transform
     
     pil_image = Image.open(image_path).convert("RGB")
     cv_img = np.array(pil_image)