Update app.py

app.py

@@ -3,7 +3,13 @@ import spacy
 from transformers import pipeline
 from PIL import Image
 
-nlp = spacy.load("en_core_web_sm")
+# Check if the model is installed, if not, download it
+try:
+    nlp = spacy.load("en_core_web_sm")
+except OSError:
+    st.write("Downloading spaCy model...")
+    subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
+    nlp = spacy.load("en_core_web_sm")
 
 st.title("SpaGAN Demo")
 st.write("Enter a text, and the system will highlight the geo-entities within it.")