hh1199 commited on
Commit
8f673a0
·
verified ·
1 Parent(s): a568576

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- #return result['labels'][0]
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,