Spaces:
Sleeping
Sleeping
Alexandre Gazola
commited on
Commit
·
446117e
1
Parent(s):
cd354b2
fix
Browse files- constants.py +2 -2
- count_max_distinct_bird_species_tool.py +4 -4
constants.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import os
|
2 |
|
3 |
-
|
4 |
-
MODEL = 'gemini-2.0-flash-exp'
|
5 |
# MODEL = 'gemini-2.5-pro-exp-03-25'
|
6 |
# 'gemini-2.5-pro-exp-03-25'
|
7 |
# gemini-2.0-flash
|
|
|
1 |
import os
|
2 |
|
3 |
+
MODEL = 'gemini-2.0-flash'
|
4 |
+
# MODEL = 'gemini-2.0-flash-exp'
|
5 |
# MODEL = 'gemini-2.5-pro-exp-03-25'
|
6 |
# 'gemini-2.5-pro-exp-03-25'
|
7 |
# gemini-2.0-flash
|
count_max_distinct_bird_species_tool.py
CHANGED
@@ -6,13 +6,13 @@ from utils import decode_base64_to_frame
|
|
6 |
@tool
|
7 |
def count_max_distinct_bird_species_in_video(youtube_url: str) -> int:
|
8 |
"""
|
9 |
-
Count the maximum number of
|
10 |
|
11 |
Args:
|
12 |
- youtube_url: str: the URL of the YouTube video
|
13 |
|
14 |
Returns:
|
15 |
-
- int: The maximum number of
|
16 |
"""
|
17 |
|
18 |
frames = sample_youtube_video(youtube_url, 5)
|
@@ -20,13 +20,13 @@ def count_max_distinct_bird_species_in_video(youtube_url: str) -> int:
|
|
20 |
|
21 |
def count_max_distinct_bird_species(framesStr: list[str]) -> int:
|
22 |
"""
|
23 |
-
Count the maximum number of
|
24 |
|
25 |
Args:
|
26 |
- frames (List[cv2.Mat]): A list of frames (images) to analyze.
|
27 |
|
28 |
Returns:
|
29 |
-
- int: The maximum number of
|
30 |
"""
|
31 |
|
32 |
# List of bird class indices from COCO dataset (class IDs for birds)
|
|
|
6 |
@tool
|
7 |
def count_max_distinct_bird_species_in_video(youtube_url: str) -> int:
|
8 |
"""
|
9 |
+
Count the maximum number of bird species in a YouTube video.
|
10 |
|
11 |
Args:
|
12 |
- youtube_url: str: the URL of the YouTube video
|
13 |
|
14 |
Returns:
|
15 |
+
- int: The maximum number of bird species detected in the YouTube video.
|
16 |
"""
|
17 |
|
18 |
frames = sample_youtube_video(youtube_url, 5)
|
|
|
20 |
|
21 |
def count_max_distinct_bird_species(framesStr: list[str]) -> int:
|
22 |
"""
|
23 |
+
Count the maximum number of bird species in a single frame considering a list of frames.
|
24 |
|
25 |
Args:
|
26 |
- frames (List[cv2.Mat]): A list of frames (images) to analyze.
|
27 |
|
28 |
Returns:
|
29 |
+
- int: The maximum number of bird species detected in any single frame.
|
30 |
"""
|
31 |
|
32 |
# List of bird class indices from COCO dataset (class IDs for birds)
|