Spaces:

pragnakalp
/

bert_based_ner

Running

pragnakalp commited on Jun 6

Commit

5f72f45

•

1 Parent(s): 73d2ad5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ def generate_ner(article):
         text += article
         # print(">>>>>>>>>>>",bert_ner_model,"<<<<<<<<<<<<<<<<<<")
         # model_ner = Ner(bert_ner_model)
-        model_ner = Ner("pytorch_model.bin")
         output = model_ner.predict(text)
         print(output)

         text += article
         # print(">>>>>>>>>>>",bert_ner_model,"<<<<<<<<<<<<<<<<<<")
         # model_ner = Ner(bert_ner_model)
+        model_ner = Ner()
         output = model_ner.predict(text)
         print(output)