Update app.py

app.py

@@ -200,7 +200,7 @@ def predict():
             'mask': mask_base64,
             'overlay': overlay_base64,
             'tumor_percentage': float(tumor_percentage),
-            'image_size': f"{original_shape[1]}x{original_shape[0]}"
+            'image_size': f"{mask.shape[1]}x{mask.shape[0]}"
         }
         
         print(f"✓ Prediction completed: {tumor_percentage:.2f}% tumor detected")