Update app.py

app.py

@@ -135,16 +135,35 @@ 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.")
-            models = []
-            for pipe in spacy.info()["pipelines"]:
-                models.append(pipe)
+            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()]
             option = streamlit.selectbox(
                 'Choose a NER model you want to apply in the list: ',
-                models)
+                models,
+                index=0)
             model = option
+            model_loaded = None
             if model != "":
-                flag_model = True
+                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()
 
             if flag_model:
                 gpu = streamlit.checkbox('Check to use GPU (if available)', value=False)
@@ -174,7 +193,7 @@ def main():
             if streamlit.button('Launch'):
                 plain = "\n".join(sentences)
                 with streamlit.spinner('Initialize NER...'):
-                    nlp = spacy.load(model)
+                    nlp = model_loaded
                     nlp.max_length = 5000000
                     if linking:
                         nlp.add_pipe('entityfishing',