amirgame197 commited on
Commit
69a2ec7
·
verified ·
1 Parent(s): e6399c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -27,6 +27,10 @@ def doo(video):
27
  # Convert the processed frame back to a NumPy array
28
  processed_frame = np.array(out)
29
 
 
 
 
 
30
  # Append the processed frame to the list
31
  processed_frames.append(processed_frame)
32
 
 
27
  # Convert the processed frame back to a NumPy array
28
  processed_frame = np.array(out)
29
 
30
+ # Ensure the processed frame has shape (height, width, 3)
31
+ if processed_frame.shape[2] != 3:
32
+ raise ValueError("Processed frame does not have 3 channels (RGB)")
33
+
34
  # Append the processed frame to the list
35
  processed_frames.append(processed_frame)
36