Avinash250325 commited on
Commit
22ff133
·
verified ·
1 Parent(s): f2ccd2c

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +7 -1
backend.py CHANGED
@@ -7,7 +7,13 @@ from PyPDF2 import PdfReader
7
  import spacy
8
 
9
  # Load SpaCy and models
10
- nlp = spacy.load("en_core_web_sm")
 
 
 
 
 
 
11
  embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
12
 
13
  tokenizer = T5Tokenizer.from_pretrained("./T5base_Question_Generation")
 
7
  import spacy
8
 
9
  # Load SpaCy and models
10
+ try:
11
+ nlp = spacy.load("en_core_web_sm")
12
+ except OSError:
13
+ from spacy.cli import download
14
+ download("en_core_web_sm")
15
+ nlp = spacy.load("en_core_web_sm")
16
+
17
  embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
18
 
19
  tokenizer = T5Tokenizer.from_pretrained("./T5base_Question_Generation")