Spaces:

spark-nlp
/

SparkNLP_NER

Build error

aemin commited on Dec 11, 2021

Commit

7f8483b

1 Parent(s): 26361d1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -180,13 +180,8 @@ def load_sparknlp_models():
                                                     .setInputCols(["sentence", "token"])\
                                                     .setOutputCol("embeddings")
-          if ner_model_name=="nerdl_fewnerd_100d":
-              model_dict[ner_model_name]= NerDLModel.pretrained(ner_model_name)\
-                                              .setInputCols(["sentence", "token", "embeddings"])\
-                                              .setOutputCol("ner")
-          else:
-              model_dict[ner_model_name]= NerDLModel.pretrained(ner_model_name, "en")\
                                               .setInputCols(["document", "token", "embeddings"])\
                                               .setOutputCol("ner")

                                                     .setInputCols(["sentence", "token"])\
                                                     .setOutputCol("embeddings")
+          model_dict[ner_model_name]= NerDLModel.pretrained(ner_model_name, "en")\
                                               .setInputCols(["document", "token", "embeddings"])\
                                               .setOutputCol("ner")