Amruthaa commited on
Commit
dde29c3
·
1 Parent(s): 386cf49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ ner_model = pipeline("ner", model = "dslim/bert-large-NER")
13
 
14
  def get_ner(text):
15
  output = ner_model(text)
16
- return {"text":input, "entities":output}
17
 
18
  demo = gr.Blocks()
19
  with demo:
 
13
 
14
  def get_ner(text):
15
  output = ner_model(text)
16
+ return {"text":text, "entities":output}
17
 
18
  demo = gr.Blocks()
19
  with demo: