Update app.py

app.py

@@ -44,7 +44,7 @@ def infer_full_vol(tensor, model):
     return output.squeeze().detach().cpu().numpy()
 
 def infer_patch_based(tensor, model, patch_size=64, stride_length=32, stride_width=32, stride_depth=16, batch_size=10, num_worker=2):
-    test_subject = tio.Subject(img = tio.ScalarImage(tensor=tensor))
+    test_subject = tio.Subject(img = tio.ScalarImage(tensor=tensor.unsqueeze(0)))
     overlap = np.subtract(patch_size, (stride_length, stride_width, stride_depth))
     
     with torch.no_grad():