Spaces:

weizmannscience
/

tokenflow

Runtime error

Linoy Tsaban commited on Sep 4, 2023

Commit

5ef9d3c

1 Parent(s): 536f1c7

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -42,7 +42,7 @@ def video_to_frames(video_path, img_size=(512,512)):
         image = T.ToPILImage()(video[i])
         # get new height and width to maintain aspect ratio
-        height, width = image.shape
         new_height = img_size[0] * height / width
         new_width  = img_size[1] * width / height

         image = T.ToPILImage()(video[i])
         # get new height and width to maintain aspect ratio
+        height, width = image.size
         new_height = img_size[0] * height / width
         new_width  = img_size[1] * width / height