Update src/streamlit_app.py

src/streamlit_app.py

@@ -9,13 +9,18 @@ import nltk
 
 # Setup NLTK and benepar
 
+    
 nltk_data_path = "/tmp/nltk_data"
 nltk.download('punkt', download_dir=nltk_data_path)
 nltk.data.path.append(nltk_data_path)
 
 benepar.download('benepar_en3')
 
-nlp = spacy.load("en_core_web_sm")
+try:
+    nlp = spacy.load("en_core_web_sm")
+except:
+    subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
+    nlp = spacy.load("en_core_web_sm")
 if "benepar" not in nlp.pipe_names:
     nlp.add_pipe("benepar", config={"model": "benepar_en3"})