danger level

app.py

@@ -228,9 +228,9 @@ def process_video(input_video, out_fps = 'auto', skip_frames = 12):
             last_5_human_detected[drawn_count % 5] = int(top_pred['human_n'] > 0)
             last_5_dangerous_dist[drawn_count % 5] = int(top_pred['dangerous_dist'] > 0)
 
-            top_pred['shark_sighted'] = int(np.sum(last_5_shark_detected) > 3)
-            top_pred['human_sighted'] = int(np.sum(last_5_human_detected) > 3)
-            top_pred['dangerous_dist_confirmed'] = int(np.sum(last_5_dangerous_dist) > 3)
+            top_pred['shark_sighted'] = int(np.sum(last_5_shark_detected) > 2)
+            top_pred['human_sighted'] = int(np.sum(last_5_human_detected) > 2)
+            top_pred['dangerous_dist_confirmed'] = int(np.sum(last_5_dangerous_dist) > 2)
 
             frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
             prediction_frame = cv2.cvtColor(display_frame, cv2.COLOR_BGR2RGB)