Bug fix

app.py

@@ -14,7 +14,7 @@ import gradio as gr
 
 model_card = ModelCard(
     name="Transkun Piano Transcription",
-    decription=("Transcribes solo piano performance into MIDI notation"),
+    description=("Transcribes solo piano performance into MIDI notation"),
     author="Yujia Yan, Zhiyao Duan",
     tags=["transcription"]
 )
@@ -61,7 +61,7 @@ def transcribe(input_file):
         waveform = librosa.resample(waveform.astype(np.float32), orig_sr=sr, target_sr=model.fs)
         sr = model.fs
 
-    x = torch.from_numpy(waveform).to(device)
+    x = torch.from_numpy(waveform.reshape(-1, 1)).to(device)
 
     notesEst = model.transcribe(x, discardSecondHalf=False)