Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ florence_processor = AutoProcessor.from_pretrained("microsoft/Florence-2-large",
|
|
| 21 |
|
| 22 |
# Load a quick, highly accurate MiDaS model pipeline via torch hub
|
| 23 |
midas = torch.hub.load("intel-isl/MiDaS", "MiDaS_small",trust_repo=True).to(device).eval()
|
| 24 |
-
midas_transforms = torch.hub.load("intel-isl/MiDaS", "transforms")
|
| 25 |
transform_depth = midas_transforms.flag_transform if hasattr(midas_transforms, 'flag_transform') else midas_transforms.dpt_transform
|
| 26 |
|
| 27 |
|
|
|
|
| 21 |
|
| 22 |
# Load a quick, highly accurate MiDaS model pipeline via torch hub
|
| 23 |
midas = torch.hub.load("intel-isl/MiDaS", "MiDaS_small",trust_repo=True).to(device).eval()
|
| 24 |
+
midas_transforms = torch.hub.load("intel-isl/MiDaS", "transforms",trust_repo=True)
|
| 25 |
transform_depth = midas_transforms.flag_transform if hasattr(midas_transforms, 'flag_transform') else midas_transforms.dpt_transform
|
| 26 |
|
| 27 |
|