Update README.md
Browse files
README.md
CHANGED
@@ -72,7 +72,8 @@ transform = transforms.Compose([
|
|
72 |
model.eval()
|
73 |
model.model.to(device)
|
74 |
y_pred= []
|
75 |
-
|
|
|
76 |
|
77 |
with torch.no_grad():
|
78 |
# in_tens = model.processor(img, return_tensors='pt')['pixel_values']
|
|
|
72 |
model.eval()
|
73 |
model.model.to(device)
|
74 |
y_pred= []
|
75 |
+
# Put your image to tes
|
76 |
+
img = Image.open('206496010652.png').convert('RGB')
|
77 |
|
78 |
with torch.no_grad():
|
79 |
# in_tens = model.processor(img, return_tensors='pt')['pixel_values']
|