Fix supporting the new color detection

.gitignore

@@ -1,4 +1,4 @@
-data/video.mp4
+assets/videos/video.mp4
 __pycache__
 .venv
 .env

example_inference.py

@@ -46,15 +46,17 @@
     # In this example, we will perform color prediction on the first frame of
     # the video
     print("5) Perform action on the media... (in this case color prediction)")
-    main_colors, _, _ = ColorPrediction.find_main_colors(image=read_first_frame('data/video.mp4'),
-                                                         min_clusters=1,
-                                                         max_clusters=5,
-                                                         downsample_factor=0.95,
-                                                         increase_elbow=0,
-                                                         verbose=False,
-                                                         plot=False)
-    print("\tMain colors found in the first frame:", main_colors.tolist())
+    ColorPrediction.find_main_colors(
+        image=read_first_frame('data/video.mp4'),
+        min_clusters=1,
+        max_clusters=8,
+        downsample_factor=0.95,
+        increase_elbow=0,
+        verbose=True,
+        plot=False)
 
     # Cleanup the video
     print("6) Cleaning up the video...")
-    os.remove('data/video.mp4')
+    os.remove('data/video.mp4')
+    print()
+

requirements.txt

@@ -1,5 +1,5 @@
 opencv-python
 python-dotenv
-uugai-python-color-prediction>=1.3.0
+uugai-python-color-prediction>=2.2.0
 uugai-python-dynamic-queue>=1.1.0
 uugai-python-kerberos-vault>=1.2.0