Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,11 +13,11 @@ logger = logging.getLogger(__name__)
|
|
| 13 |
|
| 14 |
# Load models
|
| 15 |
logger.info("Loading feature extractor model...")
|
| 16 |
-
feature_extractor = load_model(
|
| 17 |
logger.info("Feature extractor model loaded successfully.")
|
| 18 |
|
| 19 |
logger.info("Loading classification model...")
|
| 20 |
-
classification_model = joblib.load(
|
| 21 |
logger.info("Classification model loaded successfully.")
|
| 22 |
|
| 23 |
# Streamlit App
|
|
|
|
| 13 |
|
| 14 |
# Load models
|
| 15 |
logger.info("Loading feature extractor model...")
|
| 16 |
+
feature_extractor = load_model("custom_cnn_feature_extractor_finale.h5")
|
| 17 |
logger.info("Feature extractor model loaded successfully.")
|
| 18 |
|
| 19 |
logger.info("Loading classification model...")
|
| 20 |
+
classification_model = joblib.load("RF_model_finale.pkl")
|
| 21 |
logger.info("Classification model loaded successfully.")
|
| 22 |
|
| 23 |
# Streamlit App
|