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']
|