Spaces:
Sleeping
Sleeping
update app.py
Browse files
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 |
-
|
| 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 |
|