fix error for return piano roll

jdasam · Feb 18, 2021 · cbcc906 · cbcc906
1 parent 9159f99
commit cbcc906
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/transcribe.py b/transcribe.py
@@ -97,7 +97,10 @@ def inference(self, audio):
             self.switch_on_or_off()
             # time_list.append(time())
             if self.num_under_thr > self.patience:
-                return [], []
+                if self.return_roll:
+                    return [0]*88
+                else:
+                    return [], []
             self.update_mel_buffer()
             # time_list.append(time())
             acoustic_out = self.update_acoustic_out(self.mel_buffer.transpose(-1, -2))