Hugging Face's logo

Spaces:
zswwsz
/
Dissertation_txt
Runtime error

App Files Community
zswwsz commited on
Commit
4a3c3e1
•
1 Parent(s): 7006c79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import re
5
  import torch
6
 
7
  tokenizer = BertTokenizer.from_pretrained(r'./model')
8
- model = BertForSequenceClassification.from_pretrained(r'./model', num_labels = 6).device('cpu')
9
 
10
  labels = ['伤感', '快乐', '励志', '宣泄', '平静', '感人']
11
 
 
5
  import torch
6
 
7
  tokenizer = BertTokenizer.from_pretrained(r'./model')
8
+ model = BertForSequenceClassification.from_pretrained(r'./model', num_labels = 6)
9
 
10
  labels = ['伤感', '快乐', '励志', '宣泄', '平静', '感人']
11