adamtayzzz commited on
Commit
d06411a
1 Parent(s): 8994aeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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[task_name]()
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