Fix opencv installation

Dockerfile

@@ -12,10 +12,12 @@ ENV HOME=/home/user \
 WORKDIR $HOME/app
 COPY --chown=user . .
 
-RUN mamba env create --prefix $HOME/env  -f ./environment.yml
+RUN mamba env create -n app  -f ./environment.yml
 
 EXPOSE 7860
 WORKDIR $HOME/app
 
-CMD mamba run -p $HOME/env --no-capture-output voila --no-browser --port=7860 notebooks/app.ipynb
+CMD mamba run -n app --no-capture-output voila \
+  --no-browser --port=7860 --show_tracebacks=True \
+  notebooks/app.ipynb
 # CMD python3 -m voila --no-browser notebooks/ --port=7860

environment.yml

@@ -1,17 +1,24 @@
 name: voila
 channels:
   - conda-forge
+  - fastai
+  - pytorch
+  - nvidia
 dependencies:
-  - python
+  # - opencv
+  - opencv-python-headless
+  - ipywidgets
   - ipykernel
   - pip
   - ultralytics=8.0.186
   - pillow
   - python-dotenv
-  - opencv
-  - jupyterlab_widgets
   - voila
-  - ipywidgets
+  - ca-certificates
+  - certifi
+  - openssl
+  - pytorch
+  - torchvision
+  - pytorch-cuda=12.1
+  - albumentations
 
-  - pip:
-    - voila-topbar

notebooks/app.ipynb

@@ -21,7 +21,7 @@
     "\n",
     "%matplotlib inline\n",
     "\n",
-    "os.chdir('/home/camille/code/surveillance/')\n",
+    "os.chdir('..')\n",
     "_ = load_dotenv()\n"
    ]
   },
@@ -31,7 +31,12 @@
    "source": [
     "# Re-surveilling surveillance\n",
     "\n",
-    "This is a demonstration of using object detection models to locate surveillance cameras in Google Street View imagery. There are two models: **YOLO**, which runs relatively quickly but is less accurate, and **RT-DETR**, a newer model type that is slower but better at finding cameras.\n",
+    "This is a demonstration of using object detection models to locate surveillance cameras in Google Street View imagery. \n",
+    "\n",
+    "There are two models:\n",
+    "\n",
+    "- **YOLO**, which runs relatively quickly but is less accurate\n",
+    "- **RT-DETR**, a newer model type that is slower but better at finding cameras\n",
     "\n",
     "Enter a location to look up and adjust the camera parameters. After clicking the button, any cameras detected by the models will be highlighted."
    ]