sczhou commited on
Commit
5c532b1
β€’
1 Parent(s): 6aa9697
Files changed (1) hide show
  1. web-demos/hugging_face/app.py +1 -1
web-demos/hugging_face/app.py CHANGED
@@ -88,7 +88,7 @@ def get_frames_from_video(video_input, video_state):
88
  break
89
 
90
  # TODO: hard code to avoid out of memory
91
- t, h, w = len(frames), frames[0].shape[1], frames[0].shape[2]
92
  print(f'Inp video shape: t_{t}, s_{h}x_{w}')
93
  if len(frames) > 150 and max(frames[0].shape) > 1024:
94
  raise ValueError('Due to GPU memory constraints, the current version of this demo supports videos \
 
88
  break
89
 
90
  # TODO: hard code to avoid out of memory
91
+ t, h, w = len(frames), frames[0].shape[0], frames[0].shape[1]
92
  print(f'Inp video shape: t_{t}, s_{h}x_{w}')
93
  if len(frames) > 150 and max(frames[0].shape) > 1024:
94
  raise ValueError('Due to GPU memory constraints, the current version of this demo supports videos \