edichief commited on
Commit
b57c5aa
·
1 Parent(s): 7921022
Files changed (1) hide show
  1. visualize_pipeline.py +3 -0
visualize_pipeline.py CHANGED
@@ -74,6 +74,9 @@ def visualize_pipeline():
74
  import benepar
75
  load_state.markdown ("#### Downloading model...")
76
  benepar.download('benepar_en3')
 
 
 
77
  load_state.markdown ("#### Loading done!")
78
 
79
  # Pipeline
 
74
  import benepar
75
  load_state.markdown ("#### Downloading model...")
76
  benepar.download('benepar_en3')
77
+ if "model" not in st.session_state:
78
+ nlp = spacy.load("en_healthsea")
79
+ st.session_state["model"] = nlp
80
  load_state.markdown ("#### Loading done!")
81
 
82
  # Pipeline