add upload to colab

notebooks/test_model.ipynb

@@ -185,6 +185,32 @@
     "Alternatively, you can start from another audio altogether, resulting in a kind of style transfer."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "fc620a80",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "try:\n",
+    "    # are we running on Google Colab?\n",
+    "    from google.colab import files\n",
+    "    audio_file = list(files.upload().keys())[0]\n",
+    "except:\n",
+    "    audio_file = None"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ec894340",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "if audio_file is None:\n",
+    "    audio_file = \"/home/teticio/Music/Music/Cesar Mariano And CIA/Gilles Peterson In Brazil_ Da Hora/2-07 Futebol De Bar (Heavy Üsker Mix.mp3\""
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -195,7 +221,6 @@
    "outputs": [],
    "source": [
     "start_steps = 500  #@param {type:\"slider\", min:0, max:1000, step:10}\n",
-    "audio_file = \"/home/teticio/Music/Music/Cesar Mariano And CIA/Gilles Peterson In Brazil_ Da Hora/2-07 Futebol De Bar (Heavy Üsker Mix.mp3\"  #@param {type:\"string\"}\n",
     "audio_diffusion.mel.load_audio(audio_file)\n",
     "track = np.array([])\n",
     "generator = torch.Generator()\n",
@@ -212,8 +237,7 @@
     "        generator=generator)\n",
     "    display(Audio(audio, rate=sample_rate))\n",
     "    display(Audio(audio2, rate=sample_rate))\n",
-    "    track = np.concatenate([track, audio2])\n",
-    "display(Audio(track, rate=sample_rate))"
+    "    track = np.concatenate([track, audio2])"
    ]
   },
   {