Update app.py
Browse files
app.py
CHANGED
@@ -6,14 +6,13 @@ classifier = pipeline("zero-shot-classification", model="cointegrated/rubert-tin
|
|
6 |
def classify(item: str, categories: str) -> str:
|
7 |
categories_list = [c.strip() for c in categories.split(",")]
|
8 |
result = classifier(item, categories_list, multi_label=False)
|
9 |
-
return result['labels'][0]
|
10 |
-
|
11 |
-
print(classifier("Молоток"))
|
12 |
|
13 |
iface = gr.Interface(
|
14 |
fn=classify,
|
15 |
inputs=[
|
16 |
-
gr.Textbox(label="Название товара"),
|
17 |
gr.Textbox(label="Категории (через запятую)", value="Техника, Овощи, Инструменты, Морковка")
|
18 |
],
|
19 |
outputs=gr.Textbox(label="Категория")
|
|
|
6 |
def classify(item: str, categories: str) -> str:
|
7 |
categories_list = [c.strip() for c in categories.split(",")]
|
8 |
result = classifier(item, categories_list, multi_label=False)
|
9 |
+
#return result['labels'][0]
|
10 |
+
return classifier("Молоток")
|
|
|
11 |
|
12 |
iface = gr.Interface(
|
13 |
fn=classify,
|
14 |
inputs=[
|
15 |
+
gr.Textbox(label="Название товара vjkjnjr"),
|
16 |
gr.Textbox(label="Категории (через запятую)", value="Техника, Овощи, Инструменты, Морковка")
|
17 |
],
|
18 |
outputs=gr.Textbox(label="Категория")
|