Spaces:

Blealtan
/

clip-guided-binary-autoencoder

Runtime error

Blealtan commited on Feb 15, 2023

Commit

a3c8a90

1 Parent(s): 2034096

Work on arbitrary image size.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -246,8 +246,7 @@ def encode(model_prefix, img):
     encoder, _ = prepare_model(model_prefix)
     img_transform = transforms.Compose([
         transforms.PILToTensor(),
-        transforms.ConvertImageDtype(torch.float),
-        transforms.Resize((224, 224))
     ])
     with torch.no_grad():

     encoder, _ = prepare_model(model_prefix)
     img_transform = transforms.Compose([
         transforms.PILToTensor(),
+        transforms.ConvertImageDtype(torch.float)
     ])
     with torch.no_grad():