diff --git a/gemini_for_labels.py b/gemini_for_labels.py index ffab75e..859872b 100644 --- a/gemini_for_labels.py +++ b/gemini_for_labels.py @@ -210,7 +210,11 @@ for path_str in args.input_paths: labels_txt = (INPUT_DIR / "labels").read_text() valid_labels_set = set(line.strip() for line in labels_txt.splitlines() if line.strip()) -names_txt = (INPUT_DIR / "names").read_text() +names_path = (INPUT_DIR / "names") +if !os.path.exists(names_path): + names_path = Path("names") +names_txt = names_path.read_text() + valid_names_set = set(line.strip() for line in names_txt.splitlines() if line.strip()) valid_names_set.add("Unknown") valid_names_set.add("Continued")