abdullahmubeen10 commited on
Commit
b9aab1c
Β·
verified Β·
1 Parent(s): e91bb18

Update Demo.py

Browse files
Files changed (1) hide show
  1. Demo.py +2 -1
Demo.py CHANGED
@@ -49,7 +49,8 @@ def create_pipeline(model):
49
  .setInputCols(["document", "token"]) \
50
  .setOutputCol("embeddings")
51
 
52
- ner_model = NerDLModel.pretrained(model, 'es') \
 
53
  .setInputCols(["document", "token", "embeddings"]) \
54
  .setOutputCol("ner")
55
 
 
49
  .setInputCols(["document", "token"]) \
50
  .setOutputCol("embeddings")
51
 
52
+ lang = 'xx' if model == "ner_wikiner_glove_840B_300" else 'es'
53
+ ner_model = NerDLModel.pretrained(model, lang) \
54
  .setInputCols(["document", "token", "embeddings"]) \
55
  .setOutputCol("ner")
56