Spaces:

ilushado
/

article_roberta_classifier

Runtime error

ilushado commited on Apr 23, 2023

Commit

18d808e

1 Parent(s): 133387e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from torch.utils.data import Dataset, DataLoader
 from transformers import RobertaModel, RobertaTokenizer
 from transformers import AutoModel, DistilBertTokenizer
 import transformers
 idx_to_tag = {0: 'cs',
  1: 'stat',

 from transformers import RobertaModel, RobertaTokenizer
 from transformers import AutoModel, DistilBertTokenizer
 import transformers
+!pip install accelerate
 idx_to_tag = {0: 'cs',
  1: 'stat',