different resize

app.py

@@ -129,7 +129,7 @@ with gr.Blocks() as demo:
             print("Resizing image to square")
             img = Image.fromarray(img)
             transform = transforms.Compose([
-                transforms.Resize(size),
+                transforms.Resize((size, size)),
                 transforms.CenterCrop(size)
             ])
             img = transform(img)  # .transpose(1, 2, 0)
@@ -260,7 +260,7 @@ with gr.Blocks() as demo:
             img = Image.fromarray(original_image)
 
             transform = transforms.Compose([
-                transforms.Resize(224),
+                transforms.Resize((224, 224)),
                 transforms.CenterCrop(224)
             ])
             img = np.array(transform(img))