Spaces:
Runtime error
Runtime error
Update with larger max size
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ def get_transform(name: str):
|
|
50 |
if any(x in name for x in ('dino', 'mocov3', 'convnext', )):
|
51 |
normalize = transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
|
52 |
transform = transforms.Compose([
|
53 |
-
transforms.Resize(size=
|
54 |
transforms.ToTensor(),
|
55 |
normalize
|
56 |
])
|
|
|
50 |
if any(x in name for x in ('dino', 'mocov3', 'convnext', )):
|
51 |
normalize = transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
|
52 |
transform = transforms.Compose([
|
53 |
+
transforms.Resize(size=512, interpolation=TF.InterpolationMode.BICUBIC, max_size=1024),
|
54 |
transforms.ToTensor(),
|
55 |
normalize
|
56 |
])
|