Update app.py

app.py

@@ -6,8 +6,8 @@ from keras.models import load_model
 import tensorflow as tf
 from tensorflow import keras
 
-# Cargar el nuevo modelo entrenado
-model1 = load_model('./isatron_v3.h5')
+# Cargar el modelo entrenado
+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]))
-    # Convertir mapa de calor a RGB
+    # Aplicar un colormap que no contenga amarillo, como VIRIDIS
     heatmap = np.uint8(255 * heatmap)
-    heatmap = cv2.applyColorMap(heatmap, cv2.COLORMAP_JET)
+    heatmap = cv2.applyColorMap(heatmap, cv2.COLORMAP_VIRIDIS)  # Cambiado a VIRIDIS
     # Convertir imagen a BGR
     img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
     # Aplicar mapa de calor a la imagen original