Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
|
|
5 |
classifier = pipeline("text-classification", model="cointegrated/rubert-tiny2")
|
6 |
|
7 |
# Категории в порядке, соответствующем меткам модели
|
8 |
-
CATEGORIES = ["Овощи", "Инструменты", "Коробки"] # Измените на ваши категории!
|
9 |
|
10 |
def classify(item: str, categories: list) -> str:
|
11 |
prompt = f"""
|
@@ -22,7 +22,7 @@ iface = gr.Interface(
|
|
22 |
fn=classify,
|
23 |
inputs=[
|
24 |
gr.Textbox(label="Название товара 1"),
|
25 |
-
gr.Textbox(label="Категории (через запятую)", value="Овощи, Инструменты,
|
26 |
],
|
27 |
outputs=gr.Textbox(label="Категория")
|
28 |
)
|
|
|
5 |
classifier = pipeline("text-classification", model="cointegrated/rubert-tiny2")
|
6 |
|
7 |
# Категории в порядке, соответствующем меткам модели
|
8 |
+
CATEGORIES = ["Овощи", "Инструменты", "Коробки", "Прочее"] # Измените на ваши категории!
|
9 |
|
10 |
def classify(item: str, categories: list) -> str:
|
11 |
prompt = f"""
|
|
|
22 |
fn=classify,
|
23 |
inputs=[
|
24 |
gr.Textbox(label="Название товара 1"),
|
25 |
+
gr.Textbox(label="Категории (через запятую)", value="Овощи, Инструменты, Коробки, Прочее")
|
26 |
],
|
27 |
outputs=gr.Textbox(label="Категория")
|
28 |
)
|