Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,6 +61,7 @@ transformers.utils.logging.set_verbosity_error()
|
|
| 61 |
task_name = 'sst2'
|
| 62 |
model_type = 'transkimer'
|
| 63 |
|
|
|
|
| 64 |
label_list = processor.get_labels()
|
| 65 |
label_to_id = {v: i for i, v in enumerate(label_list)}
|
| 66 |
|
|
@@ -74,7 +75,6 @@ model = TranskimerForSequenceClassification.from_pretrained(model_path,from_tf=b
|
|
| 74 |
# Preprocessing the datasets
|
| 75 |
sentence1_key, sentence2_key = task_to_keys[task_name]
|
| 76 |
|
| 77 |
-
processor = processors[task_name]()
|
| 78 |
|
| 79 |
padding = False
|
| 80 |
|
|
|
|
| 61 |
task_name = 'sst2'
|
| 62 |
model_type = 'transkimer'
|
| 63 |
|
| 64 |
+
processor = processors[task_name]()
|
| 65 |
label_list = processor.get_labels()
|
| 66 |
label_to_id = {v: i for i, v in enumerate(label_list)}
|
| 67 |
|
|
|
|
| 75 |
# Preprocessing the datasets
|
| 76 |
sentence1_key, sentence2_key = task_to_keys[task_name]
|
| 77 |
|
|
|
|
| 78 |
|
| 79 |
padding = False
|
| 80 |
|