hh1199 commited on
Commit
037596a
·
verified ·
1 Parent(s): d2a154d

Update app.py

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