Spaces:

farid678
/

midas

Runtime error

farid678 commited on Dec 16, 2025

Commit

47e7e45

verified ·

1 Parent(s): ff1a9c0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from PIL import Image
 # Load MiDaS model from Torch Hub
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-midas = torch.hub.load("isl-org/MiDaS", "MiDaS_small")
 midas.to(device)
 midas.eval()

 # Load MiDaS model from Torch Hub
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+midas = torch.hub.load("isl-org/MiDaS", "MiDaS_small", trust_repo=True)
 midas.to(device)
 midas.eval()