Add stop token visualizaiotn for the notebook

2026-09-10 11:50:20 +08:00 · 2018-05-03 05:56:33 -07:00
parent f8d5bbd5d2
commit d02dd47958
2 changed files with 10 additions and 5 deletions
@@ -32,4 +32,4 @@ def plot_spectrogram(linear_output, audio):
    data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
    data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
    plt.close()
-    return data
+    return data