Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,8 @@ from PIL import Image
|
|
| 7 |
|
| 8 |
transform = transforms.Compose([
|
| 9 |
transforms.Resize((128, 128)),
|
| 10 |
-
transforms.ToTensor()
|
|
|
|
| 11 |
])
|
| 12 |
|
| 13 |
model = models.resnet18()
|
|
|
|
| 7 |
|
| 8 |
transform = transforms.Compose([
|
| 9 |
transforms.Resize((128, 128)),
|
| 10 |
+
transforms.ToTensor(),
|
| 11 |
+
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
|
| 12 |
])
|
| 13 |
|
| 14 |
model = models.resnet18()
|