EmanHassan26 commited on
Commit
579c825
·
verified ·
1 Parent(s): d256021

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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