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 |
|