momo commited on
Commit
98087f5
1 Parent(s): 1154c5d
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -15,7 +15,8 @@ model_name_list = [
15
 
16
  model = BertForSequenceClassification.from_pretrained(
17
  model_name,
18
- num_labels= 15
 
19
  )
20
 
21
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
15
 
16
  model = BertForSequenceClassification.from_pretrained(
17
  model_name,
18
+ num_labels= 15,
19
+ problem_type="multi_label_classification"
20
  )
21
 
22
  tokenizer = AutoTokenizer.from_pretrained(model_name)
requirements.txt CHANGED
@@ -1,2 +1,2 @@
1
- transformers==4.5.0
2
  torch==1.9.0
 
1
+ transformers
2
  torch==1.9.0