fixesdd
Browse files- 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
|
14 |
-
|
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}
|