Spaces:

spark-nlp
/

SparkNLP_NER

Build error

aemin commited on Dec 11, 2021

Commit

26361d1

1 Parent(s): f2af3e6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

                                                     .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")