diff --git a/hubconf.py b/hubconf.py index 42367f18..615b9a9c 100644 --- a/hubconf.py +++ b/hubconf.py @@ -1,5 +1,7 @@ dependencies = ['torch'] +import torch + from models.midas_net import MidasNet def MiDaS(pretrained=True, **kwargs):