amezi commited on
Commit
7988374
·
1 Parent(s): 91522ee
Files changed (1) hide show
  1. 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):