root commited on
Commit
33ff51e
·
1 Parent(s): bdfbee6
Files changed (2) hide show
  1. app.py +14 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -3,6 +3,8 @@ import pdfplumber
3
  import io
4
  import spacy
5
  from transformers import pipeline
 
 
6
 
7
  st.set_page_config(
8
  page_title="Resume Screener & Skill Extractor",
@@ -10,11 +12,21 @@ st.set_page_config(
10
  layout="wide"
11
  )
12
 
13
- # Load the NLP model
 
 
 
 
 
 
 
 
 
 
14
  @st.cache_resource
15
  def load_models():
16
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
17
- nlp = spacy.load("en_core_web_sm")
18
  return summarizer, nlp
19
 
20
  # Initialize models
 
3
  import io
4
  import spacy
5
  from transformers import pipeline
6
+ import subprocess
7
+ import sys
8
 
9
  st.set_page_config(
10
  page_title="Resume Screener & Skill Extractor",
 
12
  layout="wide"
13
  )
14
 
15
+ # Download spaCy model if not already downloaded
16
+ @st.cache_resource
17
+ def download_spacy_model():
18
+ try:
19
+ nlp = spacy.load("en_core_web_sm")
20
+ except OSError:
21
+ subprocess.check_call([sys.executable, "-m", "spacy", "download", "en_core_web_sm"])
22
+ nlp = spacy.load("en_core_web_sm")
23
+ return nlp
24
+
25
+ # Load the NLP models
26
  @st.cache_resource
27
  def load_models():
28
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
29
+ nlp = download_spacy_model()
30
  return summarizer, nlp
31
 
32
  # Initialize models
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  streamlit>=1.31.0
2
  pdfplumber>=0.10.3
3
  transformers>=4.37.2
 
4
  spacy>=3.7.2
5
- https://huggingface.co/spacy/en_core_web_sm/resolve/main/en_core_web_sm-any-py3-none-any.whl
 
1
  streamlit>=1.31.0
2
  pdfplumber>=0.10.3
3
  transformers>=4.37.2
4
+ torch>=2.1.2
5
  spacy>=3.7.2