hh1199 commited on
Commit
becbc70
·
verified ·
1 Parent(s): 28a81fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Загружаем модель
5
- classifier = pipeline("text-classification", model="cointegrated/rubert-tiny2")
6
 
7
  # Категории в порядке, соответствующем меткам модели
8
  CATEGORIES = ["Овощи", "Инструменты", "Коробки", "Прочее"] # Измените на ваши категории!
@@ -21,7 +21,7 @@ def classify(item: str, categories: list) -> str:
21
  iface = gr.Interface(
22
  fn=classify,
23
  inputs=[
24
- gr.Textbox(label="Название товара 1"),
25
  gr.Textbox(label="Категории (через запятую)", value="Овощи, Инструменты, Коробки, Прочее")
26
  ],
27
  outputs=gr.Textbox(label="Категория")
 
2
  from transformers import pipeline
3
 
4
  # Загружаем модель
5
+ classifier = pipeline("text-classification", model="sberbank-ai/ruRoberta-large")
6
 
7
  # Категории в порядке, соответствующем меткам модели
8
  CATEGORIES = ["Овощи", "Инструменты", "Коробки", "Прочее"] # Измените на ваши категории!
 
21
  iface = gr.Interface(
22
  fn=classify,
23
  inputs=[
24
+ gr.Textbox(label="Название товара Сбер"),
25
  gr.Textbox(label="Категории (через запятую)", value="Овощи, Инструменты, Коробки, Прочее")
26
  ],
27
  outputs=gr.Textbox(label="Категория")