SankethShetty001 commited on
Commit
f0617ea
·
verified ·
1 Parent(s): 5c19178

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -15,8 +15,14 @@ model_name = "valhalla/t5-base-qg-hl"
15
  tokenizer = T5Tokenizer.from_pretrained(model_name)
16
  model = T5ForConditionalGeneration.from_pretrained(model_name)
17
 
18
- # Load spaCy for NER
19
- nlp = spacy.load("en_core_web_sm")
 
 
 
 
 
 
20
 
21
  # Load QA pipeline
22
  qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
 
15
  tokenizer = T5Tokenizer.from_pretrained(model_name)
16
  model = T5ForConditionalGeneration.from_pretrained(model_name)
17
 
18
+ import spacy
19
+ try:
20
+ nlp = spacy.load("en_core_web_sm")
21
+ except OSError:
22
+ from spacy.cli import download
23
+ download("en_core_web_sm")
24
+ nlp = spacy.load("en_core_web_sm")
25
+
26
 
27
  # Load QA pipeline
28
  qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")