Балаганский Никита Николаевич commited on
Commit
beee43b
1 Parent(s): 11aa723

fix softmax labels

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -145,10 +145,8 @@ def main():
145
  target_label_id = 1
146
  act_type = "sigmoid"
147
  else:
148
- label2id = cls_model_config.label2id
149
- filtered_label2id = {"NEGATIVE": label2id["NEGATIVE"]}
150
- label_key = st.selectbox(ATTRIBUTE_LABEL[language], filtered_label2id.keys())
151
- target_label_id = filtered_label2id[label_key]
152
  act_type = "softmax"
153
  st.write(WARNING_TEXT[language])
154
  show_pos_alpha = st.checkbox("Show positive alphas", value=False)
 
145
  target_label_id = 1
146
  act_type = "sigmoid"
147
  else:
148
+ label_key = st.selectbox(ATTRIBUTE_LABEL[language], ["Negative"])
149
+ target_label_id = 0
 
 
150
  act_type = "softmax"
151
  st.write(WARNING_TEXT[language])
152
  show_pos_alpha = st.checkbox("Show positive alphas", value=False)