Update app.py

app.py

@@ -7,6 +7,8 @@ import torch
 tokenizer = BertTokenizer.from_pretrained(r'./model')
 model = BertForSequenceClassification.from_pretrained(r'./model', num_labels = 6)
 
+labels = ['伤感', '快乐', '励志', '宣泄', '平静', '感人']
+
 def preprocess(temp):
     temp = re.sub(u"\n\n", "\n", temp)
     temp = re.sub(u"(^\n)|(\n$)", "", temp)