paragon-analytics commited on
Commit
f057421
1 Parent(s): 6561dd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -11,6 +11,7 @@ from transformers import RobertaTokenizer, RobertaModel
11
  from transformers import AutoModelForSequenceClassification
12
  from transformers import TFAutoModelForSequenceClassification
13
  from transformers import AutoTokenizer
 
14
  # from transformers_interpret import SequenceClassificationExplainer
15
 
16
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
11
  from transformers import AutoModelForSequenceClassification
12
  from transformers import TFAutoModelForSequenceClassification
13
  from transformers import AutoTokenizer
14
+ import matplotlib.pyplot as plt
15
  # from transformers_interpret import SequenceClassificationExplainer
16
 
17
  device = "cuda:0" if torch.cuda.is_available() else "cpu"