Darayut commited on
Commit
304a554
·
verified ·
1 Parent(s): 1b0ccdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(r"../final_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(r"../final_model/RF_model_finale.pkl")
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