Update app.py

app.py

@@ -59,16 +59,16 @@ 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)
         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)
         y_transformed = destandarize_point(y0_transformed, dim=dim, flip = False)
         wrapped_img = wrap_points(img_out.detach(), x_transformed, y_transformed, l=1)
-        img_out2 = wrapped_img.detach()
+        img_out2 = wrapped_img
         marked_image = torchvision.transforms.ToPILImage()(img_out2[0])
     else:
-        #source_im_w_points = torchvision.transforms.ToPILImage()(torch.zeros(3,128,128))
+        source_im_w_points = torchvision.transforms.ToPILImage()(torch.zeros(3,128,128))
         marked_image = torchvision.transforms.ToPILImage()(torch.zeros(3,128,128))
     return [registered_img,source_im_w_points, marked_image]