ahmedxeno commited on
Commit
d1c9b62
·
1 Parent(s): b4bbe67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,12 +7,12 @@ import torch
7
  import timm
8
  import numpy as np
9
 
10
- midas = torch.hub.load('intel-isl/MiDaS', 'MiDaS_small')
11
  midas.to('cpu')
12
  midas.eval()
13
 
14
  transforms = torch.hub.load('intel-isl/MiDaS', 'transforms')
15
- transform = transforms.small_transform
16
 
17
  def predict_image(img):
18
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
 
7
  import timm
8
  import numpy as np
9
 
10
+ midas = torch.hub.load('intel-isl/MiDaS', 'DPT_Hybrid')
11
  midas.to('cpu')
12
  midas.eval()
13
 
14
  transforms = torch.hub.load('intel-isl/MiDaS', 'transforms')
15
+ transform = transforms.dpt_transform
16
 
17
  def predict_image(img):
18
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)