Wassleboss commited on
Commit
cca7013
·
1 Parent(s): 0b85e63

inshallah

Browse files
Files changed (1) hide show
  1. CnnClassification.py +4 -5
CnnClassification.py CHANGED
@@ -48,18 +48,17 @@ def load_image(img_path):
48
  img_array = img_array / 255.
49
  return img_array
50
 
51
- MODEL_PATH = os.path.join("", "models\modele_crnn_hd_final_13genres.keras")
52
- MAPPING_PATH = os.path.join("", "models\genre_mapping.json")
53
-
54
  # ==========================================
55
  # 2. CHARGEMENT DU MODÈLE ET DU MAPPING
56
  # ==========================================
57
  print("⏳ Chargement du modèle...")
58
- model_CNN = tf.keras.models.load_model("models\modele_crnn_hd_final_13genres.keras")
59
  print("✅ Modèle chargé !")
60
 
61
  print("⏳ Chargement des genres...")
62
- with open("models\genre_mapping.json", 'r') as f:
63
  mapping = json.load(f)
64
  classes = {v: k for k, v in mapping.items()}
65
  print(f"✅ {len(classes)} genres connus.")
 
48
  img_array = img_array / 255.
49
  return img_array
50
 
51
+ MODEL_PATH = os.path.join("models", "modele_crnn_hd_final_13genres.keras")
52
+ MAPPING_PATH = os.path.join("models", "genre_mapping.json")
 
53
  # ==========================================
54
  # 2. CHARGEMENT DU MODÈLE ET DU MAPPING
55
  # ==========================================
56
  print("⏳ Chargement du modèle...")
57
+ model_CNN = tf.keras.models.load_model("models/modele_crnn_hd_final_13genres.keras")
58
  print("✅ Modèle chargé !")
59
 
60
  print("⏳ Chargement des genres...")
61
+ with open(MAPPING_PATH, 'r') as f:
62
  mapping = json.load(f)
63
  classes = {v: k for k, v in mapping.items()}
64
  print(f"✅ {len(classes)} genres connus.")