fix

app.py

@@ -34,13 +34,13 @@ def main():
     cls_model_config = AutoConfig.from_pretrained(cls_model_name)
     if cls_model_config.problem_type == "multi_label_classification":
         label2id = cls_model_config.label2id
-        print(list(label2id.keys()))
-        label_key = st.selectbox("Веберите нужный атрибут текста", list(label2id.keys())[-1])
+        label_key = st.selectbox("Веберите нужный атрибут текста", label2id.keys())
         target_label_id = label2id[label_key]
     else:
         label2id = cls_model_config.label2id
-        label_key = st.selectbox("Веберите нужный атрибут текста", label2id.keys()[-1])
-        target_label_id = 0
+        print(list(label2id.keys()))
+        label_key = st.selectbox("Веберите нужный атрибут текста", [list(label2id.keys())[-1]])
+        target_label_id = 1
     prompt = st.text_input("Начало текста:", "Привет")
     alpha = st.slider("Alpha:", min_value=-10, max_value=10, step=1, value=0)
     entropy_threshold = st.slider("Entropy Threshold:", min_value=0., max_value=5., step=.1, value=0.)