zswwsz commited on
Commit
fd5d5e3
1 Parent(s): 708b8d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from sklearn.metrics.pairwise import cosine_similarity
12
  tokenizer = BertTokenizer.from_pretrained(r'./model')
13
  model = BertForSequenceClassification.from_pretrained(r'./model', num_labels = 6)
14
 
15
- labels = ['伤感', '快乐', '励志', '宣泄', '平静', '感人']
16
 
17
  def preprocess(temp):
18
  temp = re.sub(u"\n\n", "\n", temp)
 
12
  tokenizer = BertTokenizer.from_pretrained(r'./model')
13
  model = BertForSequenceClassification.from_pretrained(r'./model', num_labels = 6)
14
 
15
+ labels = ['伤感(sadness)', '快乐(happiness)', '励志(inspiration)', '宣泄(catharsis)', '平静(placid)', '感人(touching)']
16
 
17
  def preprocess(temp):
18
  temp = re.sub(u"\n\n", "\n", temp)