From eac7510a3ba90c8919ca545f641d351f8038558d Mon Sep 17 00:00:00 2001
From: Cedric Verstraeten <cedric@verstraeten.io>
Date: Wed, 7 Aug 2024 21:20:30 +0200
Subject: [PATCH] add first github action (still stupid)

---
 .github/workflows/docker-publish.yml | 24 ++++++++++++++++++++++++
 Dockerfile                           |  9 ++-------
 2 files changed, 26 insertions(+), 7 deletions(-)
 create mode 100644 .github/workflows/docker-publish.yml

diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
new file mode 100644
index 0000000..79eca6e
--- /dev/null
+++ b/.github/workflows/docker-publish.yml
@@ -0,0 +1,24 @@
+name: Build and Publish Docker Image
+on:
+  push:
+    branches:
+      - main
+jobs:
+  build-and-push:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v2
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v1
+      - name: Log in to Docker Hub
+        uses: docker/login-action@v1
+        with:
+          username: ${{ secrets.DOCKER_USERNAME }}
+          password: ${{ secrets.DOCKER_PASSWORD }}
+      - name: Build and push Docker image
+        uses: docker/build-push-action@v2
+        with:
+          context: .
+          push: true
+          tags: uugai/data-filter:latest
diff --git a/Dockerfile b/Dockerfile
index 6338ee5..ef1e4a5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -40,7 +40,7 @@ COPY . .
 ENV MEDIA_SAVEPATH "/ml/data/input/input_video.mp4"
 
 # Model parameters
-ENV MODEL_NAME "yolov8n-seg.pt"
+ENV MODEL_NAME "yolov8n.pt"
 
 # Queue parameters
 ENV QUEUE_NAME "" 
@@ -72,11 +72,6 @@ ENV RETURN_JSON_SAVEPATH "/ml/data/output/output_json.json"
 ENV TIME_VERBOSE "True"
 ENV LOGGING "True"
 
-ENV FIND_DOMINANT_COLORS "False"
-ENV COLOR_PREDICTION_INTERVAL "1"
-ENV MIN_CLUSTERS "3"
-ENV MAX_CLUSTERS "3"
-
 # Classification parameters
 ENV CLASSIFICATION_FPS ""
 ENV CLASSIFICATION_THRESHOLD ""
@@ -88,6 +83,6 @@ ENV ALLOWED_CLASSIFICATIONS "0, 1, 2, 3, 5, 7, 14, 15, 16, 24, 26, 28"
 
 
 # Run the application
-ENTRYPOINT ["python" , "object_classification_yolov8.py"]
+ENTRYPOINT ["python" , "queue_filter.py"]