lterriel commited on
Commit
1e96eb1
·
1 Parent(s): 5bc501a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -126,9 +126,10 @@ def main():
126
  if flag_view:
127
  streamlit.write("## ⚙️ Configure NER pipeline and options:")
128
  streamlit.write("⚠️ Using Bert based model and/or linking may increase considerably the processing time.")
129
- models = []
130
- for pipe in spacy.info()["pipelines"]:
131
- models.append(pipe)
 
132
  option = streamlit.selectbox(
133
  'Choose a NER model you want to apply in the list: ',
134
  models)
 
126
  if flag_view:
127
  streamlit.write("## ⚙️ Configure NER pipeline and options:")
128
  streamlit.write("⚠️ Using Bert based model and/or linking may increase considerably the processing time.")
129
+ # Normaly: Load from PIP or directory (install issues with HF spaces)
130
+ models = ["fr_ner4archives_v3_with_vectors", "fr_ner4archives_v3", "fr_ner4archives_V3_camembert_base"]
131
+ #for pipe in spacy.info()["pipelines"]:
132
+ # models.append(pipe)
133
  option = streamlit.selectbox(
134
  'Choose a NER model you want to apply in the list: ',
135
  models)