Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,7 @@ transformers.utils.logging.set_verbosity_error()
|
|
| 61 |
task_name = 'sst2'
|
| 62 |
model_type = 'transkimer'
|
| 63 |
|
| 64 |
-
processor = processors[
|
| 65 |
label_list = processor.get_labels()
|
| 66 |
label_to_id = {v: i for i, v in enumerate(label_list)}
|
| 67 |
|
|
|
|
| 61 |
task_name = 'sst2'
|
| 62 |
model_type = 'transkimer'
|
| 63 |
|
| 64 |
+
processor = processors['sst-2']()
|
| 65 |
label_list = processor.get_labels()
|
| 66 |
label_to_id = {v: i for i, v in enumerate(label_list)}
|
| 67 |
|