Spaces:

momo
/

Hate_speech_Privacy_Detection

Sleeping

momo commited on Jun 5, 2022

Commit

c164ef4

1 Parent(s): e24f20d

add app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,6 @@ model_name_list = [
 model = AutoModelForSequenceClassification.from_pretrained(
     model_name,
     num_labels=15,
-    problem_type="multi_label_classification"
 ).cuda()
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 model = AutoModelForSequenceClassification.from_pretrained(
     model_name,
     num_labels=15,
 ).cuda()
 tokenizer = AutoTokenizer.from_pretrained(model_name)