for debugging in hugginface

offline_pipeline.py

@@ -15,6 +15,7 @@ def retrieve_audio_information(
     sample_id: str,
     stt_model: str,
 ) -> tuple[Optional[str], Optional[str], Optional[str], Optional[str], Optional[str]]:
+    
     noisy_spec_path = f"/tmp/{sample_id}_noisy_spectrogram.png"
     enhanced_spec_path = f"/tmp/{sample_id}_enhanced_spectrogram.png"
     spec_image(original_array).save(noisy_spec_path)
@@ -55,7 +56,7 @@ def load_local_file(
         gr.Warning("Please upload a valid audio file.")
         raise ValueError("Missing audio sample. Please upload an audio sample or use the microphone input.")
     new_sample_stem = os.path.splitext(os.path.basename(sample_path))[0]
-    y_16k, _ = librosa.load(sample_path, sr=DEFAULT_SR, dtype="float32")
+    y_16k, _ = librosa.load(sample_path, sr=DEFAULT_SR, dtype="float32", mono=True)
     return y_16k, new_sample_stem
 
 def load_file_from_dataset(sample_id: str) -> tuple[tuple, np.ndarray, str]:
@@ -63,6 +64,10 @@ def load_file_from_dataset(sample_id: str) -> tuple[tuple, np.ndarray, str]:
         gr.Warning("Please select a sample from the dropdown.")
         raise ValueError("No sample selected. Please select a sample from the dropdown.")
     new_sample_stem = sample_id
-    y_16k, sr = get_audio(sample_id, prefix="mix")
-    y_16k_for_gradio = to_gradio_audio(y_16k, sr)# Convert to 16-bit PCM for gradio audio component
+    try:
+        y_16k, sr = get_audio(sample_id, prefix="mix")
+        y_16k_for_gradio = to_gradio_audio(y_16k, sr)
+    except Exception as e:  # Convert to 16-bit PCM for gradio audio component
+        gr.Warning(f"{e}")
+        raise e
     return y_16k_for_gradio, y_16k, new_sample_stem