better visualisation for compIAM pitch

webbook/introduction/compiam.ipynb

@@ -428,33 +428,7 @@
     "audio, sr = our_track.audio\n",
     "\n",
     "# Plotting waveform\n",
-    "pitch_frames = int(segment_in_seconds * sr * 0.01)\n",
-    "plot_pitch(\n",
-    "    pitch=predicted_pitch[:pitch_frames, 1],\n",
-    "    times=predicted_pitch[:pitch_frames, 0],\n",
-    ");\n",
-    "\n",
-    "# Playing 10 seconds of audio used for prediction\n",
-    "audio_samples = int(segment_in_seconds * sr)\n",
-    "audio_short = audio[:, :audio_samples]\n",
-    "ipd.Audio(\n",
-    "    data=audio_short,\n",
-    "    rate=sr\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from compiam.visualisation.pitch import plot_pitch\n",
-    "\n",
-    "# Plotting waveform\n",
-    "pitch_frames = int(segment_in_seconds * sr * 0.01)  ## 0.01 is the chosen step\n",
-    "pitch_values = predicted_pitch[:pitch_frames, 1]\n",
-    "pitch_values[pitch_values < 20] = None\n",
+    "pitch_frames = int(segment_in_seconds * ftanet_carnatic.sample_rate * 0.01)\n",
     "plot_pitch(\n",
     "    pitch=predicted_pitch[:pitch_frames, 1],\n",
     "    times=predicted_pitch[:pitch_frames, 0],\n",