Merge pull request #1662 from axinc-ai/multilingual-minilmv2

fix candidate_labels
axinc-ai · Mar 4, 2025 · 2ffb18e · 2ffb18e
2 parents f7724a8 + db25368
commit 2ffb18e
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