Spaces:
Running
Running
Initial commit
Browse files
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 |
|