Include package name in pip install

nestauk · Oct 31, 2024 · 0f3f3a4 · 0f3f3a4
1 parent 720aff1
commit 0f3f3a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ojd_daps_skills/extract_skills/extract_skills_utils.py b/ojd_daps_skills/extract_skills/extract_skills_utils.py
@@ -111,7 +111,7 @@ def create(
             if ner_model_name == "nestauk/en_skillner":
                 msg.info(f"{ner_model_name} NER model not loaded. Downloading model...")
                 os.system(
-                    f"pip install https://huggingface.co/{namespace}/{ner_name}/resolve/main/{ner_name}-any-py3-none-any.whl"
+                    f'pip install "{ner_name} @ https://huggingface.co/{namespace}/{ner_name}/resolve/main/{ner_name}-any-py3-none-any.whl"'
                 )
                 msg.info(f"Model downloaded")
                 nlp = spacy.load(ner_name)