Update app.py

app.py

@@ -12,7 +12,9 @@ def geometry_transform(images: list,
 
     file_names: list = [f.name for f in images]
     image_list: list = [K.io.load_image(f, K.io.ImageLoadType(0)).float().unsqueeze(0)/255 for f in file_names]
-    image_batch: torch.Tensor = torch.cat(image_list, 0)
+    if len(image_list) > 1:
+        image_list = [K.geometry.resize(x, x.shape[-2:], antialias=True) for x in image_list]
+        image_batch: torch.Tensor = torch.cat(image_list, 0)
     center: torch.Tensor = torch.tensor([x.shape[1:] for x in image_batch])/2
     translation = torch.tensor(translation).repeat(len(image_list), 2)
     scale = torch.tensor(scale).repeat(len(image_list), 2)
@@ -20,7 +22,7 @@ def geometry_transform(images: list,
     affine_matrix: torch.Tensor = KG.get_affine_matrix2d(translation, center, scale, angle)
     with torch.inference_mode():
         transformed: torch.Tensor = KG.transform.warp_affine(image_batch, affine_matrix[:, :2], dsize=image_batch.shape[2:])
-    concat_images: list = torch.cat(transformed, dim=-1)
+    concat_images: list = torch.cat([x for x in transformed], dim=-1)
     final_images: np.ndarray = K.tensor_to_image(concat_images*255).astype(np.uint8) 
 
     return final_images