amezi commited on
Commit
cd3d00e
·
1 Parent(s): 544fd4d
Files changed (1) hide show
  1. src/labeler.py +8 -2
src/labeler.py CHANGED
@@ -10,8 +10,8 @@ class TogetherLLMLabeler:
10
 
11
  def generate_label(self, game_card, transcript, spatial_context, frame_urls):
12
  prompt = f"""
13
- Game Information:
14
- {game_card}
15
 
16
  Commentary:
17
  {transcript}
@@ -37,6 +37,12 @@ class TogetherLLMLabeler:
37
  #return response.choices[0].message["content"].strip()
38
  return response.choices[0].message.content
39
 
 
 
 
 
 
 
40
  #after commentary:
41
  # Spatial Context (object detections per frame):
42
  # {spatial_context}
 
10
 
11
  def generate_label(self, game_card, transcript, spatial_context, frame_urls):
12
  prompt = f"""
13
+ Game information:
14
+ This is a football game in seria A between Juventus and Empoli. Juventus is dressed in Dark Navy kit and the keeper is in orange, Empoli, the away team is dressed in white jersey and the keeper in fluorescent yellow. The referee is dressed in a red shirt and black shorts. The home team is Juventus-- the game is being played at the Allianz stadium.
15
 
16
  Commentary:
17
  {transcript}
 
37
  #return response.choices[0].message["content"].strip()
38
  return response.choices[0].message.content
39
 
40
+
41
+ #before commentary:
42
+ # Game Information:
43
+ # {game_card}
44
+
45
+
46
  #after commentary:
47
  # Spatial Context (object detections per frame):
48
  # {spatial_context}