Update app.py

app.py

@@ -59,7 +59,7 @@ def imgnt_reg(img1,img2):#, model_selected):
         x_source = destandarize_point(x0_source, dim=dim, flip = False)
         y_source = destandarize_point(y0_source, dim=dim, flip = False)
         source_im_w_points = wrap_points(fixed_images.detach(), x_source, y_source, l=1)
-        source_im_w_points = torchvision.transforms.ToPILImage()(source_im_w_points)
+        source_im_w_points = torchvision.transforms.ToPILImage()(source_im_w_points[0])
         M_Predicted = workaround_matrix(pred['Affine_mtrx'].detach(), acc = 0.5/crop_ratio)
         x0_transformed, y0_transformed = transform_standard_points(M_Predicted[0], x0_source, y0_source)
         x_transformed = destandarize_point(x0_transformed, dim=dim, flip = False)