oceddyyy commited on
Commit
497bc17
·
verified ·
1 Parent(s): 5367515

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -7,7 +7,15 @@ import gradio as gr
7
  from huggingface_hub import Repository
8
  from datetime import datetime
9
 
10
- nlp = spacy.load("en_core_web_sm")
 
 
 
 
 
 
 
 
11
 
12
  qg_model = AutoModelForSeq2SeqLM.from_pretrained("valhalla/t5-base-qa-qg-hl")
13
  qg_tokenizer = AutoTokenizer.from_pretrained("valhalla/t5-base-qa-qg-hl", use_fast=True)
 
7
  from huggingface_hub import Repository
8
  from datetime import datetime
9
 
10
+ from spacy.cli import download
11
+
12
+ try:
13
+ nlp = spacy.load("en_core_web_sm")
14
+ except OSError:
15
+ print("Downloading 'en_core_web_sm' model...")
16
+ download("en_core_web_sm")
17
+ nlp = spacy.load("en_core_web_sm")
18
+
19
 
20
  qg_model = AutoModelForSeq2SeqLM.from_pretrained("valhalla/t5-base-qa-qg-hl")
21
  qg_tokenizer = AutoTokenizer.from_pretrained("valhalla/t5-base-qa-qg-hl", use_fast=True)