Alexandre Gazola commited on
Commit
446117e
·
1 Parent(s): cd354b2
constants.py CHANGED
@@ -1,7 +1,7 @@
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
 
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 distinct 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 distinct bird species detected in the YouTube video.
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 distinct 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 distinct bird species detected in any single frame.
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)