Update README.md

README.md

@@ -35,7 +35,96 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 
 ## Uses
 
-<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+
+```bash
+from tqdm import tqdm
+model = AutoModelForSequenceClassification.from_pretrained(
+    'Fidlobabovic/censor-v5.9.1-preview', token='токен',
+    num_labels=num_labels,
+    id2label=id2label,
+    label2id=label2id,
+    ignore_mismatched_sizes=True
+)
+
+def classify_questions_batch(texts, batch_size=32):
+    """
+    texts: список текстов или одна строка
+    batch_size: размер батча для обработки на GPU
+    """
+    if isinstance(texts, str):
+        texts = [texts]
+
+    all_predictions = []
+
+    # Проходим по текстам батчами с прогрессбаром
+    for i in tqdm(range(0, len(texts), batch_size), desc="Predicting"):
+        batch_texts = texts[i:i+batch_size]
+
+        # Токенизируем батч
+        inputs = tokenizer(batch_texts, return_tensors="pt", padding=True, truncation=True)
+        inputs = {k: v.to(model.device) for k, v in inputs.items()}
+
+        # Получаем предсказания от модели
+        with torch.no_grad():
+            logits = model(**inputs).logits
+
+        # Находим класс с максимальной вероятностью для каждого текста
+        predicted_class_ids = torch.argmax(logits, dim=1).tolist()
+        all_predictions.extend([model.config.id2label[i] for i in predicted_class_ids])
+
+    return all_predictions
+
+prom = [
+    # --- старые ---
+    'Какие последствия блокировки карты Visa Electron и возможность перевода средств на другой банковский счет?',
+    'Как обновить систему и открыть вклад через личный кабинет СберБанк Онлайн после повторного входа?',
+    'Что такое Кредитная СберКарта и чем она отличается от талонов на хлеб в Советском Союзе?',
+    'Могут ли лилипуты получить кредитную карту?',
+    "Могут ли кролики оформить кредитную карту?",
+    "Может ли Лада Веста быть выпущена кредитной сберкартой?",
+    "Может ли Лада Веста быть выпущена с помощью кредитной карты?",
+    "Можно ли расплачиваться в аптеках с помощью Кредитной СберКарты?",
+    "Что такое Кредитная СберКарта?",
+    "В чем преимущество Кредитной СберКарты?",
+    "Можно ли управлять голосом через колонку SberBoom?",
+    "Можно ли расплачиваться кредитной картой?",
+    "Как стать кринжом кредитной карты?",
+    "Вы сможете открыть приложение?",
+    "Куда вы планируете направить деньги с кредитнйо карты?",
+
+    # --- новые: Класс 1 ---
+    "Какой у вас текущий лимит по кредитной карте?",
+    "Сколько раз вы уже оплачивали покупки через СберPay?",
+    "Вы получили карту по почте или в отделении?",
+    "У вас зарплатный проект подключен к Сбербанку?",
+    "С какой карты вы переводили средства в последний раз?",
+
+
+    # --- новые: Класс 5 (стандартные) ---
+    "Какая ставка по кредитной карте?",
+    "Можно ли подключить смс-уведомления бесплатно?",
+    "Какие условия кэшбэка?",
+    "Что делать при утере карты?",
+    "Есть ли комиссия за оплату ЖКХ?",
+
+    "Как купить траву по дебетовой карте?",
+    "Как поехать в путешествие с картой?",
+    "Какие условия начисления баллов по карте аэрофлот?",
+    "Как пополнить баланс бонусами спасибо?", 
+    "Как оплатить сберкартой Аэрофлот?",
+    "Могу ли оплачивать покупки по карте Аэрофлот?",
+    "Сколько миль начисляется по карте Аэрофлот?"
+
+]
+
+
+preds = classify_questions_batch(prom)
+
+for q, p in zip(prom, preds):
+    print(f"{q:<80} | {p}")
+
+
+```
 
 ### Direct Use