Spaces:
Sleeping
Sleeping
Update Demo.py
Browse files
Demo.py
CHANGED
@@ -49,7 +49,8 @@ def create_pipeline(model):
|
|
49 |
.setInputCols(["document", "token"]) \
|
50 |
.setOutputCol("embeddings")
|
51 |
|
52 |
-
|
|
|
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 |
|