Update app.py

app.py

@@ -51,10 +51,10 @@ def imgnt_reg(img1,img2, model_selected):
     # prepare inputs:
     fixed_images = preprocess_image(img1, dim = 128)
     moving_images = preprocess_image(img2, dim = 128)
-    M_i = torch.normal(torch.zeros([2,3]), torch.ones([2,3]))
+    M_i = torch.normal(torch.zeros([1,2,3]), torch.ones([1,2,3]))
     model_inputs = {'source':fixed_images,
                 'target':moving_images,
-                'M_i' :M_i.usqueeze(0) }
+                'M_i' :M_i }
     #[moving_images/255, fixed_images/255]
     pred = IR_Model_tst(model_inputs)
     img_out = wrap_imge_cropped(pred['Affine_mtrx'], fixed_images, dim1=224, dim2=128)