changing code for video generatio

.gitignore

@@ -1,6 +1,6 @@
 # Added
 tmp/
-
+*.mp4
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

app.py

@@ -169,8 +169,8 @@ def process_video(input_video, out_fps = 'auto', skip_frames = 7):
             prediction_frame = cv2.cvtColor(display_frame, cv2.COLOR_BGR2RGB)
             
             
-            #frame = cv2.resize(frame, (int(width), int(height)))
-            video.write(cv2.cvtColor(frame, cv2.COLOR_RGB2BGR))
+            #
+            #video.write(cv2.cvtColor(frame, cv2.COLOR_RGB2BGR))
             
             if cnt*skip_frames %2==0 and top_pred['shark_sighted']: 
                 prediction_frame = cv2.resize(prediction_frame, (int(width), int(height)))
@@ -178,7 +178,9 @@ def process_video(input_video, out_fps = 'auto', skip_frames = 7):
                 
             if  top_pred['shark_sighted']: 
                 frame = draw_cockpit(frame, top_pred,cnt*skip_frames)
-                video.write(cv2.cvtColor(frame, cv2.COLOR_RGB2BGR))
+            
+            frame = cv2.resize(frame, (int(width), int(height)))
+            video.write(cv2.cvtColor(frame, cv2.COLOR_RGB2BGR))
                 
             
             pred_dashbord = prediction_dashboard(top_pred = top_pred)