Spaces:

spark-nlp
/

SparkNLP_NER

Build error

aemin commited on Dec 11, 2021

Commit

a6e3571

•

1 Parent(s): 7e1a27f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -154,7 +154,7 @@ def load_sparknlp_models():
         try:
           if embeddings_name=="glove_100d":
             model_dict[embeddings_name]= WordEmbeddingsModel.pretrained(embeddings_name, "en")\
-                                                  .setInputCols("sentence", "token")\
                                                   .setOutputCol("embeddings")
           elif embeddings_name=="elmo":

         try:
           if embeddings_name=="glove_100d":
             model_dict[embeddings_name]= WordEmbeddingsModel.pretrained(embeddings_name, "en")\
+                                                  .setInputCols(["sentence", "token"])\
                                                   .setOutputCol("embeddings")
           elif embeddings_name=="elmo":