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
|