Update app.py

app.py

@@ -7,7 +7,7 @@ import tensorflow as tf
 from tensorflow import keras
 
 # Cargar el modelo entrenado
-model1 = load_model('./isatron_v3.h5')
+model1 = load_model('./modelo_isatron_jeysshonl.h5')
 
 # Función para encontrar la última capa convolucional
 def find_last_conv_layer(model):
@@ -70,9 +70,9 @@ def make_gradcam_heatmap(img_array, model, last_conv_layer_name, pred_index=None
 def overlay_heatmap(heatmap, img, alpha=0.4):
     # Redimensionar mapa de calor al tamaño de la imagen
     heatmap = cv2.resize(heatmap, (img.shape[1], img.shape[0]))
-    # Aplicar un colormap que no contenga amarillo, como VIRIDIS
+    # Aplicar el colormap RAINBOW para obtener colores más vivos
     heatmap = np.uint8(255 * heatmap)
-    heatmap = cv2.applyColorMap(heatmap, cv2.COLORMAP_VIRIDIS)  # Cambiado a VIRIDIS
+    heatmap = cv2.applyColorMap(heatmap, cv2.COLORMAP_RAINBOW)  # Cambiado a RAINBOW
     # Convertir imagen a BGR
     img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
     # Aplicar mapa de calor a la imagen original