Spaces:
Runtime error
Runtime error
Work on arbitrary image size.
Browse files
app.py
CHANGED
@@ -246,8 +246,7 @@ def encode(model_prefix, img):
|
|
246 |
encoder, _ = prepare_model(model_prefix)
|
247 |
img_transform = transforms.Compose([
|
248 |
transforms.PILToTensor(),
|
249 |
-
transforms.ConvertImageDtype(torch.float)
|
250 |
-
transforms.Resize((224, 224))
|
251 |
])
|
252 |
|
253 |
with torch.no_grad():
|
|
|
246 |
encoder, _ = prepare_model(model_prefix)
|
247 |
img_transform = transforms.Compose([
|
248 |
transforms.PILToTensor(),
|
249 |
+
transforms.ConvertImageDtype(torch.float)
|
|
|
250 |
])
|
251 |
|
252 |
with torch.no_grad():
|