Update app.py

app.py

@@ -7,8 +7,14 @@ import gradio as gr
 from io import BytesIO
 import base64
 
-# Load pre-trained models
-nlp = spacy.load("en_core_web_sm")
+# Install the spaCy model if not already installed
+try:
+    nlp = spacy.load("en_core_web_sm")
+except OSError:
+    import subprocess
+    subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
+    nlp = spacy.load("en_core_web_sm")
+
 sent_model = SentenceTransformer('bert-base-nli-mean-tokens')
 
 def extract_entities(text):