Update app.py

app.py

@@ -68,6 +68,7 @@ def pred_maps(images):
     with torch.no_grad():
         scaled_preds_tensor = model(images_proc.to(device))[-1]
     preds = []
+    print('scaled_preds_tensor.sum():', scaled_preds_tensor.sum())
     for image_shape, pred_tensor, save_path in zip(image_shapes, scaled_preds_tensor, save_paths):
         if device == 'cuda':
             pred_tensor = pred_tensor.cpu()