Merge pull request #9 from FelipeCabelloE:master

base docker image changed to YOLO one

.devcontainer/Dockerfile

@@ -1,8 +1,7 @@
-FROM python:3.11
-RUN pip install --no-cache-dir --upgrade pip
+FROM ultralytics/ultralytics:8.0.216-cpu
 COPY requirements.txt /tmp/
 RUN pip install --quiet --no-cache-dir -r /tmp/requirements.txt
-RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
+#RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
 
 WORKDIR /workspaces/solarathon-team5/
 ENV PYTHONPATH=/workspaces/solarathon-team5/

.devcontainer/requirements.txt

@@ -1,6 +1,6 @@
 solara==1.23.0
-Pillow==10.1.0
-opencv-python==4.8.1.78
-ultralytics==8.0.215
-numpy==1.26.2
+#Pillow==10.1.0
+#opencv-python==4.8.1.78
+#ultralytics==8.0.215
+#numpy==1.26.2
 plotly==5.18.0

ploomber/Dockerfile

@@ -1,6 +1,8 @@
-FROM python:3.11
+#FROM python:3.11
+FROM ultralytics/ultralytics:8.0.216-cpu
 
-RUN pip install --no-cache-dir --upgrade pip
+#RUN pip install --no-cache-dir --upgrade pip
+#RUN pip install --pre torch torchvision --force-reinstall --index-url https://download.pytorch.org/whl/nightly/cpu
 RUN mkdir wheels
 COPY wheels/*.whl wheels
 RUN pip install wheels/*.whl

pyproject.toml

@@ -8,10 +8,10 @@ description = "Template project for Solarathon 2023"
 version = "0.0.1"
 dependencies = [
     "solara==1.23.0",
-    "Pillow==10.1.0",
-    "opencv-python==4.8.1.78",
-    "ultralytics==8.0.215",
-    "numpy==1.26.2",
+    # "Pillow==10.1.0",
+    # "opencv-python==4.8.1.78",
+    # "ultralytics==8.0.215",
+    # "numpy==1.26.2",
     "plotly==5.18.0"    
 ]