get condition from ENV

uug-ai · Aug 10, 2024 · c887eb3 · c887eb3
1 parent 479f5f5
commit c887eb3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/queue_filter.py b/queue_filter.py
@@ -127,7 +127,12 @@ def init():
             # Check if the frame_number corresponds to a frame that should be classified.
             if frame_number > 0 and frame_skip_factor > 0 and frame_number % frame_skip_factor == 0:
 
-                condition = "4 persons detected"
+                # condition = "4 persons detected"
+                condition = os.getenv("CONDITION", "")
+                if condition == "":
+                    print("No condition set, exiting")
+                    break
+
                 frame, total_time_class_prediction, conditionMet = processFrame(
                     MODEL, frame, video_out, condition)