Spaces:
Sleeping
Sleeping
all add
Browse files- app.py +2 -1
- 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
|
2 |
torch==1.9.0
|
|
|
1 |
+
transformers
|
2 |
torch==1.9.0
|