Update app.py

app.py

@@ -36,9 +36,9 @@ pipe_depth = StableDiffusionControlNetPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5", controlnet=controlnet_depth, torch_dtype=torch.float16).to("cuda")
 
 # 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
+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