Made sample number configurable to prevent container rebuild

scripts/explain.py

@@ -1,3 +1,4 @@
+import os
 import numpy as np
 from datetime import datetime
 from src.musiclime.explainer import MusicLIMEExplainer
@@ -7,16 +8,20 @@ from src.musiclime.wrapper import MusicLIMEPredictor
 def musiclime(audio_data, lyrics_text):
     """
     MusicLIME wrapper for API usage.
-
     Args:
         audio_data: Audio array (from librosa.load or similar)
         lyrics_text: String containing lyrics
-
     Returns:
         dict: Structured explanation results
     """
     start_time = datetime.now()
 
+    # Get number of samples from environment variable, default to 1000
+    num_samples = int(os.getenv("MUSICLIME_NUM_SAMPLES", "1000"))
+    num_features = int(os.getenv("MUSICLIME_NUM_FEATURES", "10"))
+
+    print(f"[MusicLIME] Using num_samples={num_samples}, num_features={num_features}")
+
     # Create musiclime instances
     explainer = MusicLIMEExplainer()
     predictor = MusicLIMEPredictor()
@@ -26,7 +31,7 @@ def musiclime(audio_data, lyrics_text):
         audio=audio_data,
         lyrics=lyrics_text,
         predict_fn=predictor,
-        num_samples=1000,
+        num_samples=num_samples,
         labels=(1,),
     )
 
@@ -35,8 +40,8 @@ def musiclime(audio_data, lyrics_text):
     predicted_class = np.argmax(original_prediction)
     confidence = float(np.max(original_prediction))
 
-    # Get top 10 features
-    top_features = explanation.get_explanation(label=1, num_features=10)
+    # Get top features (I also made this configurable to prevent rebuilding)
+    top_features = explanation.get_explanation(label=1, num_features=num_features)
 
     # Calculate runtime
     end_time = datetime.now()
@@ -68,7 +73,7 @@ def musiclime(audio_data, lyrics_text):
                 [f for f in top_features if f["type"] == "lyrics"]
             ),
             "runtime_seconds": runtime_seconds,
-            "samples_generated": 1000,
+            "samples_generated": num_samples,
             "timestamp": start_time.isoformat(),
         },
     }