NeuroSpaceX commited on
Commit
f86ae42
·
verified ·
1 Parent(s): 67d2259

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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 - Проверка на спам",