cocktailpeanut commited on
Commit
bca2d24
·
1 Parent(s): 482061c
Files changed (1) hide show
  1. demo_gradio.py +2 -0
demo_gradio.py CHANGED
@@ -35,6 +35,8 @@ from DPT.dpt.models import DPTDepthModel
35
  from DPT.dpt.midas_net import MidasNet_large
36
  from DPT.dpt.transforms import Resize, NormalizeImage, PrepareForNet
37
 
 
 
38
  """
39
  Get ZeST Ready
40
  """
 
35
  from DPT.dpt.midas_net import MidasNet_large
36
  from DPT.dpt.transforms import Resize, NormalizeImage, PrepareForNet
37
 
38
+ import devicetorch
39
+
40
  """
41
  Get ZeST Ready
42
  """