Spaces:

momo
/

Hate_speech_Privacy_Detection

Sleeping

momo commited on Jun 5, 2022

Commit

98087f5

1 Parent(s): 1154c5d

all add

Files changed (2) hide show

app.py CHANGED Viewed

@@ -15,7 +15,8 @@ model_name_list = [
 model = BertForSequenceClassification.from_pretrained(
     model_name,
-    num_labels= 15
 )
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 model = BertForSequenceClassification.from_pretrained(
     model_name,
+    num_labels= 15,
+    problem_type="multi_label_classification"
 )
 tokenizer = AutoTokenizer.from_pretrained(model_name)

requirements.txt CHANGED Viewed