bgk commited on
Commit
933d813
·
1 Parent(s): 5be9699

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def ner(text):
11
 
12
  text=text.lower()
13
 
14
- prediction, model_output = model.predict([text])
15
  filtered_output = [[{v: k} for d in sublist for k, v in d.items() if v.startswith("B-")] for sublist in prediction]
16
  entities = []
17
  for sublist in filtered_output:
 
11
 
12
  text=text.lower()
13
 
14
+ prediction= model.predict([text])
15
  filtered_output = [[{v: k} for d in sublist for k, v in d.items() if v.startswith("B-")] for sublist in prediction]
16
  entities = []
17
  for sublist in filtered_output: