Stefan Dumitrescu commited on
Commit
766d339
·
1 Parent(s): ff08140
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ elif input_method == "Write or Paste New Text":
39
 
40
  @st.cache(allow_output_mutation=True)
41
  def setModel(named_persons_only):
42
- ner = roner.NER(named_persons_only=named_persons_only)
43
  return ner
44
 
45
  @st.cache(allow_output_mutation=True)
 
39
 
40
  @st.cache(allow_output_mutation=True)
41
  def setModel(named_persons_only):
42
+ ner = roner.NER(named_persons_only=True if named_persons_only.startswith("Proper") else False)
43
  return ner
44
 
45
  @st.cache(allow_output_mutation=True)