Update app.py

app.py

@@ -24,6 +24,7 @@ ENTITY_COLORS = {
     "ORG": "#f39c12",
     "PER": "#3498db"
 }
+
 MAP_MODELS = {
     "":"",
     "fr_ner4archives_V3_camembert_base": "https://huggingface.co/ner4archives/fr_ner4archives_V3_camembert_base/resolve/main/fr_ner4archives_V3_camembert_base-any-py3-none-any.whl#sha256=919aca257e4b2342a3773e7dcc3442b30f2ac50c4a3466d5ecda5e0e337ea396",
@@ -31,6 +32,7 @@ MAP_MODELS = {
     "fr_ner4archives_v3_with_vectors":"https://huggingface.co/ner4archives/fr_ner4archives_v3_with_vectors/resolve/main/fr_ner4archives_v3_with_vectors-any-py3-none-any.whl#sha256=4c938618ce84a919008d17822980cfd914f9ad8f73231a64cd1ffcaa80ab09f2"
 }
 
+
 # Read configuration
 with open(CONFIG_FILE, mode="r") as json_file:
     CONFIGURATION = json.loads(json_file.read())
@@ -134,33 +136,15 @@ def main():
         if flag_view:
             streamlit.write("## ⚙️ Configure NER pipeline and options:")
             streamlit.write("⚠️ Using Bert based model and/or linking may increase considerably the processing time.")
-            # Normaly: Load from PIP or directory (install issues with HF spaces)
-            models = [str(key) for key in MAP_MODELS.keys()]
+            models = []
+            for pipe in spacy.info()["pipelines"]:
+                models.append(pipe)
             option = streamlit.selectbox(
                 'Choose a NER model you want to apply in the list: ',
-                models,
-                index=0)
+                models)
             model = option
-            model_loaded = None
             if model != "":
-                try:
-                    model_loaded = spacy.load(model)
-                    flag_model = True
-                    streamlit.write(f"{model} is available locally.")
-                except:
-                    placeholder = streamlit.empty()
-                    button = streamlit.button(f"Download model: {model}")
-                    with placeholder.container():
-                        if button:
-                            streamlit.write(f"Download model: {model} in progress...")
-                            p1 = subprocess.Popen(["pip", "install", MAP_MODELS[model]])
-                            o = p1.wait()
-                            if o == 0:
-                                streamlit.write(f"Download model: {model} done.")
-                                model_loaded = spacy.load(model)
-                                flag_model = True
-                                streamlit.write(f"{model} is available locally.")
-                    placeholder.empty()
+                flag_model = True
 
             if flag_model:
                 gpu = streamlit.checkbox('Check to use GPU (if available)', value=False)
@@ -190,7 +174,7 @@ def main():
             if streamlit.button('Launch'):
                 plain = "\n".join(sentences)
                 with streamlit.spinner('Initialize NER...'):
-                    nlp = model_loaded
+                    nlp = spacy.load(model)
                     nlp.max_length = 5000000
                     if linking:
                         nlp.add_pipe('entityfishing',