seanpedrickcase commited on
Commit
991fcdc
·
1 Parent(s): 9336026

Capitalise zero shot topics on input

Browse files
Files changed (1) hide show
  1. tools/llm_api_call.py +1 -1
tools/llm_api_call.py CHANGED
@@ -1000,7 +1000,7 @@ def llm_query(file_data:pd.DataFrame,
1000
  if candidate_topics:
1001
  # 'Zero shot topics' are those supplied by the user
1002
  zero_shot_topics = read_file(candidate_topics.name)
1003
- zero_shot_topics_series = zero_shot_topics.iloc[:, 0].str.strip().str.lower()
1004
  # Max 150 topics allowed
1005
  if len(zero_shot_topics_series) > 120:
1006
  print("Maximum 120 topics allowed to fit within large language model context limits.")
 
1000
  if candidate_topics:
1001
  # 'Zero shot topics' are those supplied by the user
1002
  zero_shot_topics = read_file(candidate_topics.name)
1003
+ zero_shot_topics_series = zero_shot_topics.iloc[:, 0].str.strip().str.lower().str.capitalize()
1004
  # Max 150 topics allowed
1005
  if len(zero_shot_topics_series) > 120:
1006
  print("Maximum 120 topics allowed to fit within large language model context limits.")