Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
19 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
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")
|