Update app.py

app.py

@@ -18,8 +18,8 @@ def geometry_transform(images: list,
     scale = torch.tensor(scale).repeat(len(image_list), 2)
     angle = torch.tensor(angle).repeat(len(image_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:])
+    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)
     final_images: np.ndarray = K.tensor_to_image(concat_images*255).astype(np.uint8)