3v324v23 commited on
Commit
b086439
1 Parent(s): dc291f8

Initial commit

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -16,10 +16,14 @@ import os
16
  def start_server():
17
  os.system("uvicorn InferenceServer:app --port 8080 --host 0.0.0.0 --workers 1")
18
 
 
 
 
 
 
19
  def load_models():
20
  if not is_port_in_use(8080):
21
  with st.spinner(text="Loading models, please wait..."):
22
- os.system("python -m spacy download en_core_web_sm")
23
  proc = Process(target=start_server, args=(), daemon=True)
24
  proc.start()
25
  while not is_port_in_use(8080):
@@ -183,6 +187,7 @@ if __name__ == "__main__":
183
  if not st.session_state['models_loaded']:
184
  load_models()
185
 
 
186
  nlp = spacy.load('en_core_web_sm')
187
  annotator = errant.load('en', nlp)
188
 
 
16
  def start_server():
17
  os.system("uvicorn InferenceServer:app --port 8080 --host 0.0.0.0 --workers 1")
18
 
19
+ @st.cache
20
+ def load_spacy():
21
+ os.system("python -m spacy download en_core_web_sm")
22
+
23
+
24
  def load_models():
25
  if not is_port_in_use(8080):
26
  with st.spinner(text="Loading models, please wait..."):
 
27
  proc = Process(target=start_server, args=(), daemon=True)
28
  proc.start()
29
  while not is_port_in_use(8080):
 
187
  if not st.session_state['models_loaded']:
188
  load_models()
189
 
190
+ load_spacy()
191
  nlp = spacy.load('en_core_web_sm')
192
  annotator = errant.load('en', nlp)
193