paragon-analytics
commited on
Commit
•
f057421
1
Parent(s):
6561dd9
Update app.py
Browse files
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"
|