Update app.py

app.py

@@ -3,8 +3,6 @@ import tensorflow as tf
 import numpy as np
 import cv2
 from tensorflow.keras.layers import DepthwiseConv2D
-import matplotlib.pyplot as plt
-import io
 from PIL import Image
 
 # Función personalizada para DepthwiseConv2D
@@ -41,6 +39,10 @@ labels = [
 ]
 
 def generate_heatmap(image, prediction):
+    # Asegurarse de que image sea un array de NumPy
+    if not isinstance(image, np.ndarray):
+        image = np.array(image)
+
     # Crear un mapa de calor basado en la predicción
     heatmap = np.mean(prediction, axis=-1)  # Promediar los canales de predicción
     heatmap = np.maximum(heatmap, 0)  # Aplicar ReLU