mono audio case handled

app.py

@@ -6,7 +6,12 @@ import gradio as gr
 
 def preprocess(signal, sr, device):
     
-    signal = torch.permute(signal, (1, 0))
+    # making sure channel dimension is in the 1st axis
+    if len(signal.shape) > 1:
+        signal = torch.permute(signal, (1, 0))
+    # Otherwise, add a channel dimension
+    else:
+        signal = signal.unsqueeze(0)
     
     # resampling the audio signal with the training sample rate
     if sr != 8_000:
@@ -50,4 +55,4 @@ article = "<p style='text-align: center'><a href='https://github.com/khalidsaifu
 examples = [
     ['sample_audio.wav']
 ]
-gr.Interface(pipeline, "audio", outputs, title=title, description=description, article=article, examples=examples).launch()
+gr.Interface(pipeline, "audio", outputs, title=title, description=description, article=article, examples=examples).launch(share=True)