Update Sentinel-2 Colab setup

Author: wrignj08

README.md

@@ -25,7 +25,7 @@ OmniWaterMask is a Python library for high accuracy water segmentation in high t
 
 [![Colab_Button]][Link]
 
-[Link]: https://colab.research.google.com/drive/15qgTgk4XiWcSzllNt4CtFU7pDSNrKLHk?usp=sharing 'Try OmniWaterMask In Colab'
+[Link]: https://colab.research.google.com/github/DPIRD-DMA/OmniWaterMask/blob/main/examples/Sentinel%202%20example.ipynb 'Try OmniWaterMask In Colab'
 
 [Colab_Button]: https://img.shields.io/badge/Try%20in%20Colab-grey?style=for-the-badge&logo=google-colab
 

examples/Sentinel 2 example.ipynb

@@ -1,5 +1,21 @@
 {
  "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "try:\n",
+    "    import google.colab  # type: ignore # noqa: F401\n",
+    "\n",
+    "    %pip install -q omniwatermask==0.5 s2mosaic\n",
+    "    %pip uninstall -q -y torch torchvision torchaudio\n",
+    "    %pip install -q --index-url https://download.pytorch.org/whl/cu124 torch torchvision torchaudio\n",
+    "except ImportError:\n",
+    "    pass"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,