Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -180,8 +180,13 @@ def load_sparknlp_models():
|
|
180 |
.setInputCols(["sentence", "token"])\
|
181 |
.setOutputCol("embeddings")
|
182 |
|
183 |
-
|
184 |
-
|
|
|
|
|
|
|
|
|
|
|
185 |
.setInputCols(["document", "token", "embeddings"])\
|
186 |
.setOutputCol("ner")
|
187 |
|
|
|
180 |
.setInputCols(["sentence", "token"])\
|
181 |
.setOutputCol("embeddings")
|
182 |
|
183 |
+
if ner_model_name=="nerdl_fewnerd_100d":
|
184 |
+
model_dict[ner_model_name]= NerDLModel.pretrained(ner_model_name)\
|
185 |
+
.setInputCols(["sentence", "token", "embeddings"])\
|
186 |
+
.setOutputCol("ner")
|
187 |
+
|
188 |
+
else:
|
189 |
+
model_dict[ner_model_name]= NerDLModel.pretrained(ner_model_name, "en")\
|
190 |
.setInputCols(["document", "token", "embeddings"])\
|
191 |
.setOutputCol("ner")
|
192 |
|