fix candidate_labels

axinc-ai · Mar 3, 2025 · db25368 · db25368
1 parent f7724a8
commit db25368
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/natural_language_processing/multilingual-minilmv2/multilingual-minilmv2.py b/natural_language_processing/multilingual-minilmv2/multilingual-minilmv2.py
@@ -92,7 +92,7 @@ def main():
     check_and_download_models(WEIGHT_PATH, MODEL_PATH, REMOTE_PATH)
     ailia_model = ailia.Net(MODEL_PATH, WEIGHT_PATH, env_id=args.env_id)
 
-    candidate_labels = re.split(r'\s*,\s*', CANDIDATE_LABELS) # Delete spaces before and after commas
+    candidate_labels = re.split(r'\s*,\s*', args.candidate_labels) # Delete spaces before and after commas
 
     if args.disable_ailia_tokenizer:
         from transformers import AutoTokenizer