Sara-Adjo commited on
Commit
eb1412c
Β·
verified Β·
1 Parent(s): bff4d9e

Update predict.py

Browse files
Files changed (1) hide show
  1. predict.py +1 -1
predict.py CHANGED
@@ -66,7 +66,7 @@ def predict_pytorch(image_path: str, model_path: str = "sara_model.pth") -> dict
66
  # TensorFlow Prediction β€” corrigΓ© pour incompatibilitΓ© Keras 2.x / 3.x
67
  # ════════════════════════════════════════════════════════════════════════════
68
 
69
- def predict_tensorflow(image_path: str, model_path: str = "sara_model.keras") -> dict:
70
  import tensorflow as tf
71
 
72
  # ── Couche Dense patchΓ©e qui ignore quantization_config ───────────────
 
66
  # TensorFlow Prediction β€” corrigΓ© pour incompatibilitΓ© Keras 2.x / 3.x
67
  # ════════════════════════════════════════════════════════════════════════════
68
 
69
+ def predict_tensorflow(image_path: str, model_path: str = "sara_model.h5") -> dict:
70
  import tensorflow as tf
71
 
72
  # ── Couche Dense patchΓ©e qui ignore quantization_config ───────────────