Spaces:
Runtime error
Runtime error
srivatsavdamaraju
commited on
Commit
•
511adcb
1
Parent(s):
f2befd0
Update app.py
Browse files
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",
|
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
|