Spaces:
Sleeping
Sleeping
Added resizing of imgs
Browse files
app.py
CHANGED
|
@@ -63,6 +63,7 @@ model = load_model(model_repo_id)
|
|
| 63 |
# 3. Define how to transform image
|
| 64 |
transforms = torchvision.transforms.Compose(
|
| 65 |
[
|
|
|
|
| 66 |
torchvision.transforms.ToTensor()
|
| 67 |
])
|
| 68 |
|
|
|
|
| 63 |
# 3. Define how to transform image
|
| 64 |
transforms = torchvision.transforms.Compose(
|
| 65 |
[
|
| 66 |
+
torchvision.transforms.Resize((224, 224)),
|
| 67 |
torchvision.transforms.ToTensor()
|
| 68 |
])
|
| 69 |
|