akdeniz27 commited on
Commit
edbb9d6
1 Parent(s): 8a8244f

tner xlm-r model added

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -21,7 +21,8 @@ st.title("Demo for Turkish NER Models")
21
  model_list = ['akdeniz27/bert-base-turkish-cased-ner',
22
  'akdeniz27/convbert-base-turkish-cased-ner',
23
  'akdeniz27/xlm-roberta-base-turkish-ner',
24
- 'xlm-roberta-large-finetuned-conll03-english']
 
25
 
26
  st.sidebar.header("Select NER Model")
27
  model_checkpoint = st.sidebar.radio("", model_list)
@@ -35,7 +36,7 @@ st.sidebar.header("Select Aggregation Strategy Type")
35
  if model_checkpoint == "akdeniz27/xlm-roberta-base-turkish-ner":
36
  aggregation = st.sidebar.radio("", ('simple', 'none'))
37
  st.sidebar.write(xlm_agg_strategy_info)
38
- elif model_checkpoint == "xlm-roberta-large-finetuned-conll03-english":
39
  aggregation = st.sidebar.radio("", ('simple', 'none'))
40
  st.sidebar.write(xlm_agg_strategy_info)
41
  st.sidebar.write("")
 
21
  model_list = ['akdeniz27/bert-base-turkish-cased-ner',
22
  'akdeniz27/convbert-base-turkish-cased-ner',
23
  'akdeniz27/xlm-roberta-base-turkish-ner',
24
+ 'xlm-roberta-large-finetuned-conll03-english',
25
+ 'tner/tner-xlm-roberta-base-ontonotes5']
26
 
27
  st.sidebar.header("Select NER Model")
28
  model_checkpoint = st.sidebar.radio("", model_list)
 
36
  if model_checkpoint == "akdeniz27/xlm-roberta-base-turkish-ner":
37
  aggregation = st.sidebar.radio("", ('simple', 'none'))
38
  st.sidebar.write(xlm_agg_strategy_info)
39
+ elif model_checkpoint == "xlm-roberta-large-finetuned-conll03-english" or model_checkpoint == "tner/tner-xlm-roberta-base-ontonotes5":
40
  aggregation = st.sidebar.radio("", ('simple', 'none'))
41
  st.sidebar.write(xlm_agg_strategy_info)
42
  st.sidebar.write("")