autonomous019 commited on
Commit
4625865
1 Parent(s): 88c5ee7

adding matplotlib

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -2,6 +2,8 @@ from transformers import ViTConfig, ViTForImageClassification
2
  from transformers import ViTFeatureExtractor
3
  from PIL import Image
4
  import requests
 
 
5
 
6
 
7
  # option 1: load with randomly initialized weights (train from scratch)
 
2
  from transformers import ViTFeatureExtractor
3
  from PIL import Image
4
  import requests
5
+ import matplotlib.pyplot as plt
6
+
7
 
8
 
9
  # option 1: load with randomly initialized weights (train from scratch)