fixesd
Browse files- src/labeler.py +3 -2
src/labeler.py
CHANGED
@@ -32,8 +32,9 @@ class TogetherLLMLabeler:
|
|
32 |
messages=[{"role": "user", "content": content}],
|
33 |
#max_tokens=200
|
34 |
)
|
35 |
-
|
36 |
-
return response.choices[0].message["content"].strip()
|
|
|
37 |
|
38 |
#after commentary:
|
39 |
# Spatial Context (object detections per frame):
|
|
|
32 |
messages=[{"role": "user", "content": content}],
|
33 |
#max_tokens=200
|
34 |
)
|
35 |
+
print(response.choices[0].message.content)
|
36 |
+
#return response.choices[0].message["content"].strip()
|
37 |
+
return response.choices[0].message.content
|
38 |
|
39 |
#after commentary:
|
40 |
# Spatial Context (object detections per frame):
|