Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,9 +36,9 @@ pipe_depth = StableDiffusionControlNetPipeline.from_pretrained(
|
|
| 36 |
"runwayml/stable-diffusion-v1-5", controlnet=controlnet_depth, torch_dtype=torch.float16).to("cuda")
|
| 37 |
|
| 38 |
# Load a quick, highly accurate MiDaS model pipeline via torch hub
|
| 39 |
-
midas = torch.hub.load("intel-isl/MiDaS", "MiDaS_small").to(device).eval()
|
| 40 |
-
midas_transforms = torch.hub.load("intel-isl/MiDaS", "transforms")
|
| 41 |
-
transform_depth = midas_transforms.flag_transform if hasattr(midas_transforms, 'flag_transform') else midas_transforms.dpt_transform
|
| 42 |
|
| 43 |
|
| 44 |
@spaces.GPU
|
|
|
|
| 36 |
"runwayml/stable-diffusion-v1-5", controlnet=controlnet_depth, torch_dtype=torch.float16).to("cuda")
|
| 37 |
|
| 38 |
# Load a quick, highly accurate MiDaS model pipeline via torch hub
|
| 39 |
+
#midas = torch.hub.load("intel-isl/MiDaS", "MiDaS_small").to(device).eval()
|
| 40 |
+
#midas_transforms = torch.hub.load("intel-isl/MiDaS", "transforms")
|
| 41 |
+
#transform_depth = midas_transforms.flag_transform if hasattr(midas_transforms, 'flag_transform') else midas_transforms.dpt_transform
|
| 42 |
|
| 43 |
|
| 44 |
@spaces.GPU
|