Upload folder using huggingface_hub

Dockerfile

@@ -9,6 +9,7 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 COPY requirements.txt .
+RUN pip install --no-cache-dir torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cpu
 RUN pip install --no-cache-dir -r requirements.txt
 
 COPY handler.py .

handler.py

@@ -21,6 +21,22 @@ import warnings
 warnings.filterwarnings("ignore")
 
 
+# ──────────────────────────────────────────────────────────────────────── #
+# Constants & Defaults
+# ──────────────────────────────────────────────────────────────────────── #
+
+DEFAULT_AUDIO_FEATURES = {
+    "v1_snr": 0.0,
+    "v2_noise_traffic": 0.0, "v2_noise_office": 0.0,
+    "v2_noise_crowd": 0.0, "v2_noise_wind": 0.0, "v2_noise_clean": 1.0,
+    "v3_speech_rate": 0.0,
+    "v4_pitch_mean": 0.0, "v5_pitch_std": 0.0,
+    "v6_energy_mean": 0.0, "v7_energy_std": 0.0,
+    "v8_pause_ratio": 0.0, "v9_avg_pause_dur": 0.0, "v10_mid_pause_cnt": 0,
+    "v11_emotion_stress": 0.0, "v12_emotion_energy": 0.0, "v13_emotion_valence": 0.0,
+}
+
+
 # ──────────────────────────────────────────────────────────────────────── #
 # Emotion CNN (mirrors src/emotion_features.py EmotionCNN)
 # ──────────────────────────────────────────────────────────────────────── #
@@ -281,17 +297,9 @@ async def global_exception_handler(request: Request, exc: Exception):
 
 extractor = AudioFeatureExtractorEndpoint()
 
-DEFAULT_AUDIO_FEATURES = {
-    "v1_snr": 0.0,
-    "v2_noise_traffic": 0.0, "v2_noise_office": 0.0,
-    "v2_noise_crowd": 0.0, "v2_noise_wind": 0.0, "v2_noise_clean": 1.0,
-    "v3_speech_rate": 0.0,
-    "v4_pitch_mean": 0.0, "v5_pitch_std": 0.0,
-    "v6_energy_mean": 0.0, "v7_energy_std": 0.0,
-    "v8_pause_ratio": 0.0, "v9_avg_pause_dur": 0.0, "v10_mid_pause_cnt": 0,
-    "v11_emotion_stress": 0.0, "v12_emotion_energy": 0.0, "v13_emotion_valence": 0.0,
-}
-
+# ──────────────────────────────────────────────────────────────────────── #
+# Constants & Defaults
+# ──────────────────────────────────────────────────────────────────────── #
 
 class AudioBase64Request(BaseModel):
     audio_base64: str = ""

requirements.txt

@@ -5,9 +5,8 @@ numpy==1.24.3
 scipy==1.11.2
 
 # ML
-torch==2.1.0
-torchvision==0.16.0
-scikit-image==0.22.0
+
+
 matplotlib==3.8.2
 
 # API