srivatsavdamaraju commited on
Commit
511adcb
1 Parent(s): f2befd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import threading
10
  import queue
11
 
12
  # Load the MiDaS model from PyTorch Hub
13
- model = torch.hub.load("intel-isl/MiDaS", "MiDaS_small", force_reload=True)
14
  model.eval()
15
 
16
  # Image transformation function
 
10
  import queue
11
 
12
  # Load the MiDaS model from PyTorch Hub
13
+ model = torch.hub.load("intel-isl/MiDaS", "MiDaS_small", trust_repo=True)
14
  model.eval()
15
 
16
  # Image transformation function