Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,8 @@ TOKEN = os.getenv("HF_TOKEN")
|
|
10 |
models = {
|
11 |
"ruSpamNS_v13": "NeuroSpaceX/ruSpamNS_v13",
|
12 |
"ruSpamNS_big": "NeuroSpaceX/ruSpamNS_big",
|
13 |
-
"ruSpamNS_small": "NeuroSpaceX/ruSpamNS_small"
|
|
|
14 |
}
|
15 |
|
16 |
tokenizers = {name: AutoTokenizer.from_pretrained(path, use_auth_token=TOKEN) for name, path in models.items()}
|
@@ -46,7 +47,7 @@ iface = gr.Interface(
|
|
46 |
fn=classify_text,
|
47 |
inputs=[
|
48 |
gr.Textbox(lines=3, placeholder="Введите текст..."),
|
49 |
-
gr.Radio(["ruSpamNS_v13", "ruSpamNS_big", "ruSpamNS_small"], label="Выберите модель")
|
50 |
],
|
51 |
outputs="text",
|
52 |
title="ruSpamNS - Проверка на спам",
|
|
|
10 |
models = {
|
11 |
"ruSpamNS_v13": "NeuroSpaceX/ruSpamNS_v13",
|
12 |
"ruSpamNS_big": "NeuroSpaceX/ruSpamNS_big",
|
13 |
+
"ruSpamNS_small": "NeuroSpaceX/ruSpamNS_small",
|
14 |
+
"ruSpamNS_v14": "NeuroSpaceX/ruSpamNS_v14" # Добавлена новая модель
|
15 |
}
|
16 |
|
17 |
tokenizers = {name: AutoTokenizer.from_pretrained(path, use_auth_token=TOKEN) for name, path in models.items()}
|
|
|
47 |
fn=classify_text,
|
48 |
inputs=[
|
49 |
gr.Textbox(lines=3, placeholder="Введите текст..."),
|
50 |
+
gr.Radio(["ruSpamNS_v13", "ruSpamNS_big", "ruSpamNS_small", "ruSpamNS_v14"], label="Выберите модель") # Добавлена новая модель в выбор
|
51 |
],
|
52 |
outputs="text",
|
53 |
title="ruSpamNS - Проверка на спам",
|