Update app.py
Browse files
app.py
CHANGED
@@ -6,8 +6,8 @@ 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 |
-
|
10 |
-
return classifier("Молоток")
|
11 |
|
12 |
iface = gr.Interface(
|
13 |
fn=classify,
|
|
|
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,
|