PolyakovK commited on
Commit
7b6b7c6
·
verified ·
1 Parent(s): 374bd73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,14 +4,14 @@ from transformers import pipeline
4
  classifier = pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli")
5
 
6
  def classify_email(text):
7
- labels = ["встреча", "отказ", "уточнение"]
8
  result = classifier(text, candidate_labels=labels)
9
  label = result["labels"][0]
10
- if label == "встреча":
11
  return "1"
12
- elif label == "отказ":
13
  return "2"
14
- elif label == "уточнение":
15
  return "3"
16
  else:
17
  return "?"
 
4
  classifier = pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli")
5
 
6
  def classify_email(text):
7
+ labels = ["клиент готов встретиться или хочется назначить время для звонка или встречи", "клиент отказывается от встречи, ему сейчас не интересно то, что мы предлагаем или у него нет времени", "клиент задает любые уточняющие вопросы и ему нужна помощь"]
8
  result = classifier(text, candidate_labels=labels)
9
  label = result["labels"][0]
10
+ if label == "клиент готов встретиться или хочется назначить время для звонка или встречи":
11
  return "1"
12
+ elif label == "клиент отказывается от встречи, ему сейчас не интересно то, что мы предлагаем или у него нет времени":
13
  return "2"
14
+ elif label == "клиент задает любые уточняющие вопросы и ему нужна помощь":
15
  return "3"
16
  else:
17
  return "?"