Update app.py
Browse files
app.py
CHANGED
@@ -14,6 +14,7 @@ def classify(item: str, categories: list) -> str:
|
|
14 |
К какой категории относится товар? Ответь только названием категории.
|
15 |
"""
|
16 |
result = classifier(prompt, truncation=True)
|
|
|
17 |
label_index = int(result[0]['label'].split("_")[1])
|
18 |
return CATEGORIES[label_index]
|
19 |
|
@@ -21,7 +22,7 @@ def classify(item: str, categories: list) -> str:
|
|
21 |
iface = gr.Interface(
|
22 |
fn=classify,
|
23 |
inputs=[
|
24 |
-
gr.Textbox(label="Название товара cointegrated/rubert-tiny2"),
|
25 |
gr.Textbox(label="Категории (через запятую)", value="Техника, Овощи, Инструменты, Коробки, Прочее")
|
26 |
],
|
27 |
outputs=gr.Textbox(label="Категория")
|
|
|
14 |
К какой категории относится товар? Ответь только названием категории.
|
15 |
"""
|
16 |
result = classifier(prompt, truncation=True)
|
17 |
+
print(result)
|
18 |
label_index = int(result[0]['label'].split("_")[1])
|
19 |
return CATEGORIES[label_index]
|
20 |
|
|
|
22 |
iface = gr.Interface(
|
23 |
fn=classify,
|
24 |
inputs=[
|
25 |
+
gr.Textbox(label="Название товара cointegrated/rubert-tiny2 +"),
|
26 |
gr.Textbox(label="Категории (через запятую)", value="Техника, Овощи, Инструменты, Коробки, Прочее")
|
27 |
],
|
28 |
outputs=gr.Textbox(label="Категория")
|