Update app.py

app.py

@@ -3,7 +3,7 @@ import kornia as K
 import numpy as np
 import kornia.geometry as KG
 import torch
-import cv2
+
 
 def geometry_transform(images: list,
                        translation: float,
@@ -20,8 +20,8 @@ def geometry_transform(images: list,
     affine_matrix: torch.Tensor = KG.get_affine_matrix2d(translation, center, scale, angle)
 
     transformed: torch.Tensor = KG.transform.warp_affine(image_batch, affine_matrix[:, :2], dsize=image_batch.shape[2:])
-    np_images: list = [K.tensor_to_image(f*255).astype(np.uint8) for f in transformed]
-    final_images: np.ndarray = cv2.hconcat(np_images)
+    concat_images: list = torch.cat(transformed, dim=-1)
+    final_images: np.ndarray = K.tensor_to_image(concat_images*255).astype(np.uint8) 
 
     return final_images