NIIHAAD commited on
Commit
3a6d078
·
1 Parent(s): f6f3163

update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -354,6 +354,9 @@ def predict_with_metadata(url):
354
  le_ar = music_avg_rating_le
355
  sound_type = "Music"
356
 
 
 
 
357
  # Supprimer les colonnes qui ne faisaient pas partie du training
358
  cols_to_remove = ["avg_rating", "num_downloads_class"]
359
  df_for_model = df_processed.drop(columns=[c for c in cols_to_remove if c in df_processed.columns])
@@ -365,8 +368,7 @@ def predict_with_metadata(url):
365
  pred_avg_rating_enc = model_ar.predict(df_for_model)[0]
366
  pred_avg_rating = le_ar.inverse_transform([pred_avg_rating_enc])[0]
367
 
368
- # 3️ Prétraitement seulement si durée ok
369
- df_processed = preprocess_sound(df_raw)
370
 
371
 
372
 
 
354
  le_ar = music_avg_rating_le
355
  sound_type = "Music"
356
 
357
+ # 3️ Prétraitement seulement si durée ok
358
+ df_processed = preprocess_sound(df_raw)
359
+
360
  # Supprimer les colonnes qui ne faisaient pas partie du training
361
  cols_to_remove = ["avg_rating", "num_downloads_class"]
362
  df_for_model = df_processed.drop(columns=[c for c in cols_to_remove if c in df_processed.columns])
 
368
  pred_avg_rating_enc = model_ar.predict(df_for_model)[0]
369
  pred_avg_rating = le_ar.inverse_transform([pred_avg_rating_enc])[0]
370
 
371
+
 
372
 
373
 
374