Update app.py

app.py

@@ -102,6 +102,7 @@ def process_video(input_video):
 
         # Convert the resized frame to grayscale
         gray_frame = cv2.cvtColor(resized_frame, cv2.COLOR_BGR2GRAY)
+        # Convert grayscale to 3 channels (RGB format)
         input_frame = cv2.merge([gray_frame, gray_frame, gray_frame])  # Convert grayscale to 3 channels
 
         results = model.predict(input_frame)
@@ -123,6 +124,12 @@ def process_video(input_video):
 
         # Ensure the frame is in RGB format before appending to the writer
         rgb_frame = cv2.cvtColor(resized_frame, cv2.COLOR_BGR2RGB)
+        
+        # Ensure the frame has the correct shape (640x640x3)
+        if rgb_frame.shape[2] != 3:
+            print(f"Warning: Unexpected frame shape {rgb_frame.shape}, adjusting...")
+            rgb_frame = cv2.cvtColor(resized_frame, cv2.COLOR_BGR2RGB)
+
         writer.append_data(rgb_frame)
 
     cap.release()