LoloSemper commited on
Commit
95a8fa7
·
verified ·
1 Parent(s): fa2b4a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -19,7 +19,13 @@ if not os.path.exists("saved_model"):
19
  zip_ref.extractall(".")
20
 
21
  # Cargar modelo ISIC con TensorFlow
22
- model_isic = tf.keras.models.load_model("saved_model")
 
 
 
 
 
 
23
 
24
  # Cargar modelos fastai
25
  model_malignancy = load_learner("modelo_malignancy.pkl")
 
19
  zip_ref.extractall(".")
20
 
21
  # Cargar modelo ISIC con TensorFlow
22
+ from keras.layers import TFSMLayer
23
+
24
+ try:
25
+ model_isic = TFSMLayer("saved_model", call_endpoint="serving_default")
26
+ except Exception as e:
27
+ print("🔴 Error al cargar el modelo ISIC con TFSMLayer:", e)
28
+ raise
29
 
30
  # Cargar modelos fastai
31
  model_malignancy = load_learner("modelo_malignancy.pkl")