Hugging Face's logo

Spaces:
momo
/
Hate_speech_Privacy_Detection
Sleeping

App Files Community
momo commited on
Commit
2a1fcfd
·
1 Parent(s): 50d79ba

add app

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -16,6 +16,7 @@ model_name_list = [
16
  model = BertForSequenceClassification.from_pretrained(
17
  model_name,
18
  num_labels=15,
 
19
  )
20
  tokenizer = AutoTokenizer.from_pretrained(model_name)
21
 
 
16
  model = BertForSequenceClassification.from_pretrained(
17
  model_name,
18
  num_labels=15,
19
+ problem_type="multi_label_classification"
20
  )
21
  tokenizer = AutoTokenizer.from_pretrained(model_name)
22