KrishGoyani commited on
Commit
604c992
·
verified ·
1 Parent(s): d65aeb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def ner(text, labels, threshold) :
51
  "score": 0,
52
  }
53
  for entity in model.predict_entities(
54
- text, labels, flat_ner=False, threshold=threshold
55
  )
56
  ],
57
  }
 
51
  "score": 0,
52
  }
53
  for entity in model.predict_entities(
54
+ text, labels, flat_ner=True, threshold=threshold
55
  )
56
  ],
57
  }