Fucius commited on
Commit
37b48eb
·
verified ·
1 Parent(s): a0f924e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,6 +3,7 @@ import torch
3
  torch.jit.script = lambda f: f
4
 
5
  from download import OMG_download
 
6
 
7
  import sys
8
  sys.path.append('./')
@@ -698,7 +699,7 @@ if __name__ == '__main__':
698
  'height': height,
699
  'width': width,
700
  }
701
- download = OMG_download()
702
  device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
703
  main(device, args.segment_type)
704
 
 
3
  torch.jit.script = lambda f: f
4
 
5
  from download import OMG_download
6
+ download = OMG_download()
7
 
8
  import sys
9
  sys.path.append('./')
 
699
  'height': height,
700
  'width': width,
701
  }
702
+
703
  device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
704
  main(device, args.segment_type)
705