vitaly commited on
Commit
b21e79e
·
1 Parent(s): f84f6d2

loading the model from HF hub

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -2
gradio_app.py CHANGED
@@ -6,8 +6,7 @@ from spacy import displacy
6
  from bib_tokenizers import create_references_tokenizer
7
 
8
 
9
- nlp = None
10
- nlp = spacy.load("spacy-pipelines/model-best")
11
  # return score for each token:
12
  # with threshold set to zero each suggested span is returned, and span == token,
13
  # because suggester is configured to suggest spans with len(span) == 1:
 
6
  from bib_tokenizers import create_references_tokenizer
7
 
8
 
9
+ nlp = spacy.load("en_bib_references_trf")
 
10
  # return score for each token:
11
  # with threshold set to zero each suggested span is returned, and span == token,
12
  # because suggester is configured to suggest spans with len(span) == 1: