Update app.py

app.py

@@ -6,7 +6,6 @@ import base64
 import io
 from fastai.vision.all import *
 import tensorflow as tf
-from tensorflow import keras
 import zipfile
 import os
 import traceback
@@ -17,12 +16,10 @@ if not os.path.exists("saved_model"):
         zip_ref.extractall("saved_model")
 
 # Cargar modelo ISIC con TensorFlow desde el directorio correcto
-from keras.layers import TFSMLayer
-
 try:
-    model_isic = TFSMLayer("saved_model/saved_model", call_endpoint="serving_default")
+    model_isic = tf.keras.models.load_model("saved_model/saved_model")
 except Exception as e:
-    print("\U0001F534 Error al cargar el modelo ISIC con TFSMLayer:", e)
+    print("\U0001F534 Error al cargar el modelo ISIC con tf.keras.models.load_model:", e)
     raise
 
 # Cargar modelos fastai
@@ -69,9 +66,7 @@ def analizar_lesion_combined(img):
         pred_fast_type, _, probs_fast_type = model_norm2000.predict(img_fastai)
 
         x_isic = preprocess_image_isic(img)
-        preds_isic_dict = model_isic(x_isic)
-        key = list(preds_isic_dict.keys())[0]
-        preds_isic = preds_isic_dict[key].numpy()[0]
+        preds_isic = model_isic.predict(x_isic)[0]
         pred_idx_isic = int(np.argmax(preds_isic))
         pred_class_isic = CLASSES[pred_idx_isic]
         confidence_isic = preds_isic[pred_idx_isic]
@@ -137,3 +132,4 @@ demo = gr.Interface(
 if __name__ == "__main__":
     demo.launch()
 
+