Update preprocessing.py

preprocessing.py

@@ -18,7 +18,6 @@ def read_video(video_base64, num_frames=24, target_size=(224, 224)):
     
     sampled_frames = sample_frames(frames, num_frames)
     processed_frames = pad_and_resize(sampled_frames, target_size)
-    processed_frames = processed_frames.permute(1, 0, 2, 3)  # (T, C, H, W) -> (C, T, H, W)
     return processed_frames
 
 def sample_frames(frames, num_frames):
@@ -43,6 +42,6 @@ def pad_and_resize(frames, target_size):
         transforms.ToTensor()
     ])
     processed_frames = [transform(frame) for frame in frames]
-
-    return torch.stack(processed_frames).unsqueeze(0)  # Add batch dimension
+    processed_frames = torch.stack(processed_frames)
+    return processed_frames.permute(1, 0, 2, 3).unsqueeze(0)  # Add batch dimension and permute