Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -113,7 +113,7 @@ transformer = TransformerNetwork().to(device)
|
|
| 113 |
transformer.eval()
|
| 114 |
|
| 115 |
transform = transforms.Compose([
|
| 116 |
-
|
| 117 |
transforms.ToTensor(),
|
| 118 |
transforms.Normalize(mean, std),
|
| 119 |
])
|
|
|
|
| 113 |
transformer.eval()
|
| 114 |
|
| 115 |
transform = transforms.Compose([
|
| 116 |
+
transforms.Resize(512),
|
| 117 |
transforms.ToTensor(),
|
| 118 |
transforms.Normalize(mean, std),
|
| 119 |
])
|