Spaces:
Sleeping
Sleeping
Update predict.py
Browse files- 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.
|
| 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 βββββββββββββββ
|