Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
# Загружаем модель
|
5 |
-
classifier = pipeline("text-classification", model="
|
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="Название товара
|
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="Категория")
|