freealise commited on
Commit
4696931
1 Parent(s): a66ea80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -75,6 +75,10 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
75
  # os.makedirs(outdir, exist_ok=True)
76
 
77
  for k, filename in enumerate(filenames):
 
 
 
 
78
  print('Progress {:}/{:},'.format(k+1, len(filenames)), 'Processing', filename)
79
 
80
  raw_video = cv2.VideoCapture(filename)
 
75
  # os.makedirs(outdir, exist_ok=True)
76
 
77
  for k, filename in enumerate(filenames):
78
+ file_size = os.path.getsize(filename)*1024*1024
79
+ if file_size > 128:
80
+ print(f'File size of {filename} larger than 128Mb, sorry!')
81
+ return filename
82
  print('Progress {:}/{:},'.format(k+1, len(filenames)), 'Processing', filename)
83
 
84
  raw_video = cv2.VideoCapture(filename)