hh1199 commited on
Commit
a568576
·
verified ·
1 Parent(s): 1515adb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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="Категория")