Spaces:
Sleeping
Sleeping
adamtayzzz
commited on
Commit
•
d06411a
1
Parent(s):
8994aeb
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 |
|