bgk commited on
Commit
934b8a3
·
1 Parent(s): 10a0a8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def ner(text):
19
  text=text.lower()
20
 
21
  prediction, model_output = model.predict([text])
22
- entities = prediction
23
  filtered_output = (({v: k} for d in sublist for k, v in d.items() if (v.startswith("B-") or v.startswith("I-"))) for sublist in prediction)
24
  entities = []
25
  for sublist in filtered_output:
 
19
  text=text.lower()
20
 
21
  prediction, model_output = model.predict([text])
22
+
23
  filtered_output = (({v: k} for d in sublist for k, v in d.items() if (v.startswith("B-") or v.startswith("I-"))) for sublist in prediction)
24
  entities = []
25
  for sublist in filtered_output: