Update Simple Image Classification
Browse files
Simple Image Classification
CHANGED
@@ -10,7 +10,7 @@ feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16
|
|
10 |
model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224')
|
11 |
|
12 |
# Load and display the image
|
13 |
-
url = "
|
14 |
image = Image.open(requests.get(url, stream=True).raw)
|
15 |
plt.imshow(image)
|
16 |
plt.show()
|
|
|
10 |
model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224')
|
11 |
|
12 |
# Load and display the image
|
13 |
+
url = "https://cdn.akamai.steamstatic.com/steam/apps/821880/header.jpg?t=1652241767"
|
14 |
image = Image.open(requests.get(url, stream=True).raw)
|
15 |
plt.imshow(image)
|
16 |
plt.show()
|