Update app.py

app.py

@@ -88,8 +88,13 @@ def analizar_lesion_combined(img):
     # TensorFlow model prediction
     probs_tf = predict_tf(img)
     pred_idx_tf = int(np.argmax(probs_tf))
-    pred_class_tf = CLASSES[pred_idx_tf]
     confidence_tf = probs_tf[pred_idx_tf]
+    
+    if pred_idx_tf < len(CLASSES):
+        pred_class_tf = CLASSES[pred_idx_tf]
+    else:
+        pred_class_tf = f"Clase desconocida (índice {pred_idx_tf})"
+
 
     # Gráfico ViT
     colors_bars = [RISK_LEVELS[i]['color'] for i in range(7)]