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