Update app.py

app.py

@@ -2,6 +2,8 @@ import spaces
 import torch
 torch.jit.script = lambda f: f
 
+from download import OMG_download
+
 import sys
 sys.path.append('./')
 import argparse
@@ -693,7 +695,7 @@ if __name__ == '__main__':
         'height': height,
         'width': width,
     }
-
+    download = OMG_download()
     device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
     main(device, args.segment_type)