Update app.py

app.py

@@ -15,11 +15,22 @@ from monai.transforms import (
     Compose,
     LoadImage,
     ScaleIntensity,
+    # NUEVA IMPORTACIÓN
+    Resize, 
 )
 
+# 2. DEFINICIÓN DE TRANSFORMACIONES MEJORADAS
+# Redimensionamos explícitamente a 64x64, un tamaño común para MedNIST, 
+# y nos aseguramos de que el cargador maneje la conversión a escala de grises ('L').
 test_transforms = Compose(
-        [LoadImage(image_only=True), EnsureChannelFirst(), ScaleIntensity()]
-    )
+    [
+        LoadImage(image_only=True, convert_to_tensor=True), # Asegura que se carga como tensor
+        EnsureChannelFirst(),                               # Asegura la forma [C, H, W]
+        Resize(spatial_size=(64, 64)),                      # REDIMENSIONAMIENTO CLAVE
+        ScaleIntensity(),                                   # Normaliza
+    ]
+)
+
 
 class_names = [
     'AbdomenCT', 'BreastMRI', 'CXR', 'ChestCT', 'Hand', 'HeadCT'