Update app.py

app.py

@@ -92,10 +92,13 @@ def gradio_image_segmentation(fileobj, learn, reorder, resample, save_dir, view)
 
     save_fn = 'pred_' + img_path.stem
     save_path = save_dir / save_fn
+    
+    # Ensure only_tensor is set to False to get all values
     org_img, input_img, org_size = med_img_reader(img_path_str, 
                                                   reorder=reorder,
                                                   resample=resample,
-                                                  only_tensor=False)
+                                                  only_tensor=False,
+                                                  dtype=torch.Tensor)
     
     mask_data = inference(learn, reorder=reorder, resample=resample,
                           org_img=org_img, input_img=input_img,