Update app.py

app.py

@@ -2,11 +2,22 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
-model_path = "RUSpam/spam_deberta_v4"
-tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForSequenceClassification.from_pretrained(model_path)
+models = {
+    "RUSpam/spam_deberta_v4": "RUSpam/spam_deberta_v4",
+    "RUSpam/spamNS_v1": "RUSpam/spamNS_v1"
+}
+
+tokenizers = {}
+model_instances = {}
+
+for name, path in models.items():
+    tokenizers[name] = AutoTokenizer.from_pretrained(path)
+    model_instances[name] = AutoModelForSequenceClassification.from_pretrained(path)
+
+def predict_spam(text, model_choice):
+    tokenizer = tokenizers[model_choice]
+    model = model_instances[model_choice]
 
-def predict_spam(text):
     inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=256)
     with torch.no_grad():
         outputs = model(**inputs)
@@ -21,11 +32,13 @@ def predict_spam(text):
     
     return result
 
-    
 # Создание интерфейса Gradio
 iface = gr.Interface(
     fn=predict_spam,
-    inputs=gr.Textbox(lines=5, label="Введите текст"),
+    inputs=[
+        gr.Textbox(lines=5, label="Введите текст"),
+        gr.Radio(choices=list(models.keys()), label="Выберите модель", value="RUSpam/spam_deberta_v4")
+    ],
     outputs=[
         gr.Label(label="Результат")
     ],