fix

app.py

@@ -6,6 +6,8 @@ import streamlit as st
 import torch
 
 import transformers
+
+from transformers import AutoConfig
 import tokenizers
 
 from sampling import CAIFSampler, TopKWithTemperatureSampler
@@ -18,14 +20,18 @@ def main():
     st.header("CAIF")
     cls_model_name = st.selectbox(
         'Выберите модель классификации',
-        ('tinkoff-ai/response-quality-classifier-tiny', 'tinkoff-ai/response-quality-classifier-base',
-         'tinkoff-ai/response-quality-classifier-large', "SkolkovoInstitute/roberta_toxicity_classifier")
+        (
+            'tinkoff-ai/response-quality-classifier-tiny',
+            'tinkoff-ai/response-quality-classifier-base',
+            'tinkoff-ai/response-quality-classifier-large',
+            "SkolkovoInstitute/roberta_toxicity_classifier"
+        )
     )
     lm_model_name = st.selectbox(
         'Выберите языковую модель',
         ('sberbank-ai/rugpt3small_based_on_gpt2',)
     )
-    cls_model_config = transformers.AutoConfig.from_pretrained(cls_model_name)
+    cls_model_config = AutoConfig.from_pretrained(cls_model_name)
     if cls_model_config.problem_type == "multi_label_classification":
         label2id = cls_model_config.label2id
         label_key = st.selectbox("Веберите нужный атрибут текста", label2id.keys())