Update predict.py

predict.py

@@ -1,9 +1,3 @@
-"""
-predict.py
-----------
-Charge un modèle sauvegardé et prédit la classe d'une image.
-Corrigé pour gérer les incompatibilités de version Keras.
-"""
 
 import argparse
 import numpy as np
@@ -63,10 +57,10 @@ def predict_pytorch(image_path: str, model_path: str = "sara_model.pth") -> dict
 
 
 # ════════════════════════════════════════════════════════════════════════════
-#  TensorFlow Prediction — corrigé pour incompatibilité Keras 2.x / 3.x
+#  TensorFlow Prediction
 # ════════════════════════════════════════════════════════════════════════════
 
-def predict_tensorflow(image_path: str, model_path: str = "sara_model.h5") -> dict:
+def predict_tensorflow(image_path: str, model_path: str = "sara_model.keras") -> dict:
     import tensorflow as tf
     import numpy as np