Update app.py

app.py

@@ -15,7 +15,7 @@ models = {
     "ruSpamNS_v14_multiclass": "NeuroSpaceX/ruSpamNS_v14_multiclass",
     "ruSpamNS_v16_multiclass": "NeuroSpaceX/ruSpamNS_v16_multiclass",
     "ruSpamNS_v17_multiclass": "NeuroSpaceX/ruSpamNS_v17_multiclass",
-    "ruSpamNS_v18_multiclass": "NeuroSpaceX/ruSpamNS_v18_multiclass"
+    "ruSpamNS_v19_multiclass": "NeuroSpaceX/ruSpamNS_v19_multiclass"
 }
 
 tokenizers = {name: AutoTokenizer.from_pretrained(path, use_auth_token=TOKEN) for name, path in models.items()}
@@ -44,7 +44,7 @@ def classify_text(text, model_choice):
         outputs = model(input_ids, attention_mask=attention_mask).logits
         if "multiclass" in model_choice:
             probabilities = torch.softmax(outputs, dim=1).cpu().numpy()[0]
-            if model_choice == "ruSpamNS_v18_multiclass":
+            if model_choice == "ruSpamNS_v19_multiclass":
                 labels = ["НЕ СПАМ", "СПАМ", "НЕДВИЖИМОСТЬ/ТОВАРЫ", "ВАКАНСИИ", "РЕКЛАМА/УСЛУГИ"]
             else:
                 labels = ["НЕ СПАМ", "СПАМ", "НЕДВИЖИМОСТЬ", "ВАКАНСИИ", "ПРОДАЖА"]