Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|