Hugging Face's logo

Spaces:
momo
/
Hate_speech_Privacy_Detection
Sleeping

App Files Community
momo commited on
Commit
1154c5d
•
1 Parent(s): faf7c96

add app

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,8 @@ model_name_list = [
14
  ]
15
 
16
  model = BertForSequenceClassification.from_pretrained(
17
- model_name
 
18
  )
19
 
20
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
14
  ]
15
 
16
  model = BertForSequenceClassification.from_pretrained(
17
+ model_name,
18
+ num_labels= 15
19
  )
20
 
21
  tokenizer = AutoTokenizer.from_pretrained(model_name)