added in-painting example

notebooks/test_model.ipynb

@@ -171,7 +171,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "11f91ad3",
+   "id": "993dac2a",
    "metadata": {},
    "source": [
     "### Generate continuations (\"out-painting\")"
@@ -180,7 +180,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "756d7af5",
+   "id": "22d526e3",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -277,6 +277,33 @@
     "display(Audio(track, rate=sample_rate))"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "bf63425e",
+   "metadata": {},
+   "source": [
+    "### Fill the gap (\"in-painting\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "79b52754",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "slice = 3  #@param {type:\"integer\"}\n",
+    "audio = mel.get_audio_slice(slice)\n",
+    "_, (sample_rate,\n",
+    "    audio2) = audio_diffusion.generate_spectrogram_and_audio_from_audio(\n",
+    "        raw_audio=mel.get_audio_slice(slice),\n",
+    "        generator=generator,\n",
+    "        mask_start_secs=1,\n",
+    "        mask_end_secs=1)\n",
+    "display(Audio(audio, rate=sample_rate))\n",
+    "display(Audio(audio2, rate=sample_rate))"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "ef54cef3",
@@ -330,17 +357,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "df112a72",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "len(audio) / mel.hop_length"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ad467206",
+   "id": "a4f313f2",
    "metadata": {},
    "outputs": [],
    "source": []