hh1199 commited on
Commit
28a81fd
·
verified ·
1 Parent(s): 51b4dcc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
  classifier = pipeline("text-classification", model="cointegrated/rubert-tiny2")
6
 
7
  # Категории в порядке, соответствующем меткам модели
8
- CATEGORIES = ["Овощи", "Инструменты", "Коробки"] # Измените на ваши категории!
9
 
10
  def classify(item: str, categories: list) -> str:
11
  prompt = f"""
@@ -22,7 +22,7 @@ iface = gr.Interface(
22
  fn=classify,
23
  inputs=[
24
  gr.Textbox(label="Название товара 1"),
25
- gr.Textbox(label="Категории (через запятую)", value="Овощи, Инструменты, Коробки")
26
  ],
27
  outputs=gr.Textbox(label="Категория")
28
  )
 
5
  classifier = pipeline("text-classification", model="cointegrated/rubert-tiny2")
6
 
7
  # Категории в порядке, соответствующем меткам модели
8
+ CATEGORIES = ["Овощи", "Инструменты", "Коробки", "Прочее"] # Измените на ваши категории!
9
 
10
  def classify(item: str, categories: list) -> str:
11
  prompt = f"""
 
22
  fn=classify,
23
  inputs=[
24
  gr.Textbox(label="Название товара 1"),
25
+ gr.Textbox(label="Категории (через запятую)", value="Овощи, Инструменты, Коробки, Прочее")
26
  ],
27
  outputs=gr.Textbox(label="Категория")
28
  )