Update app and requirements

app.py

@@ -8,6 +8,7 @@ from collections import Counter
 import joblib
 import freesound
 import gensim.downloader as api
+from huggingface_hub import hf_hub_download
 
 # -------- FreeSound API --------
 client = freesound.FreesoundClient()
@@ -48,29 +49,63 @@ music_onehot_tags = joblib.load("music/music_onehot_tags.joblib")
 
 # -------- MODELS --------
 
-# Music
+# =============================
+# Load ML models at runtime
+# =============================
+
+os.makedirs("models_cache", exist_ok=True)
+
+# ---- MUSIC MODELS ----
 music_model_num_downloads = joblib.load(
-    "models/music/music_model_num_downloads.joblib"
+    hf_hub_download(
+        repo_id="NIIHAAD/freesound-models",
+        filename="music_model_num_downloads.joblib",
+        cache_dir="models_cache"
+    )
 )
+
 music_model_avg_rating = joblib.load(
-    "models/music/music_xgb_avg_rating.joblib"
+    hf_hub_download(
+        repo_id="NIIHAAD/freesound-models",
+        filename="music_xgb_avg_rating.joblib",
+        cache_dir="models_cache"
+    )
 )
+
 music_avg_rating_le = joblib.load(
-    "models/music/music_xgb_avg_rating_label_encoder.joblib"
+    hf_hub_download(
+        repo_id="NIIHAAD/freesound-models",
+        filename="music_xgb_avg_rating_label_encoder.joblib",
+        cache_dir="models_cache"
+    )
 )
 
-# EffectSound
+# ---- EFFECT SOUND MODELS ----
 effect_model_num_downloads = joblib.load(
-    "models/effectSound/effectSound_model_num_downloads.joblib"
+    hf_hub_download(
+        repo_id="NIIHAAD/freesound-models",
+        filename="effectSound_model_num_downloads.joblib",
+        cache_dir="models_cache"
+    )
 )
 effect_model_avg_rating = joblib.load(
-    "models/effectSound/effectSound_xgb_avg_rating.joblib"
+    hf_hub_download(
+        repo_id="NIIHAAD/freesound-models",
+        filename="effectSound_xgb_avg_rating.joblib",
+        cache_dir="models_cache"
+    )
 )
+
 effect_avg_rating_le = joblib.load(
-    "models/effectSound/effectSound_xgb_avg_rating_label_encoder.joblib"
+    hf_hub_download(
+        repo_id="NIIHAAD/freesound-models",
+        filename="effectSound_xgb_avg_rating_label_encoder.joblib",
+        cache_dir="models_cache"
+    )
 )
 
 
+
 # EffectSound
 scaler_samplerate_effect = joblib.load("effectSound/scaler_effectSamplerate.joblib")
 scaler_age_days_effect = joblib.load("effectSound/scaler_effectSound_age_days_log.joblib")

requirements.txt

@@ -1,8 +1,10 @@
+huggingface_hub
 gradio==6.5.0
 scikit-learn
 numpy
 pandas
 gensim
 pytz
+xgboost
 git+https://github.com/MTG/freesound-python