Merge branch 'master' into sam2_docker

Author: micaelakaplan

.github/workflows/build.yml

@@ -13,6 +13,7 @@ env:
   DOCKER_BUILD_CONFIG_BRANCH: "master"
   DOCKER_BUILD_CONFIG_PATH: ".github/docker-build-config.yml"
   DOCKER_EXAMPLES_DIRECTORY: "label_studio_ml/examples"
+  SLACK_NOTIFICATION_CHANNEL_ID: "CKL2D6D2P"
 
 jobs:
   build-image:
@@ -81,7 +82,7 @@ jobs:
           echo "image_branch_version=$image_branch_version" >> $GITHUB_OUTPUT
 
       - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3.8.0
+        uses: docker/setup-buildx-action@v3.9.0
 
       - name: Login to DockerHub
         if: ${{ !github.event.pull_request.head.repo.fork }}
@@ -151,3 +152,24 @@ jobs:
             const docker_build_config = yaml.load(docker_build_config_content);
             
             core.setOutput("matrix-include", docker_build_config);
+
+  notification:
+    name: "Send Slack Notification"
+    if: failure()
+    runs-on: ubuntu-latest
+    needs:
+      - build-image
+    steps:
+      - name: Send Slack Notification
+        if: always()
+        uses: slackapi/[email protected]
+        env:
+          WORKFLOW: "<https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }}|workflow>"
+        with:
+          method: chat.postMessage
+          token: ${{ secrets.SLACK_LSE_BOT_TOKEN }}
+          payload: |
+            channel: "${{ env.SLACK_NOTIFICATION_CHANNEL_ID }}"
+            text: >+
+              :x: Build workflow failed
+              [ ${{ env.WORKFLOW }} ]

.github/workflows/tests.yml

@@ -27,6 +27,7 @@ env:
   DOCKER_BUILD_CONFIG_BRANCH: "master"
   DOCKER_BUILD_CONFIG_PATH: ".github/docker-build-config.yml"
   DOCKER_EXAMPLES_DIRECTORY: "label_studio_ml/examples"
+  SLACK_NOTIFICATION_CHANNEL_ID: "CKL2D6D2P"
 
 jobs:
   run_pytest:
@@ -241,3 +242,24 @@ jobs:
         env:
           PR_URL: ${{ github.event.pull_request.html_url }}
           GITHUB_TOKEN: ${{ secrets.GIT_PAT }}
+
+  notification:
+    name: "Send Slack Notification"
+    if: failure() && github.event_name == 'schedule'
+    runs-on: ubuntu-latest
+    needs:
+      - run_pytest
+    steps:
+      - name: Send Slack Notification
+        if: always()
+        uses: slackapi/[email protected]
+        env:
+          WORKFLOW: "<https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }}|workflow>"
+        with:
+          method: chat.postMessage
+          token: ${{ secrets.SLACK_LSE_BOT_TOKEN }}
+          payload: |
+            channel: "${{ env.SLACK_NOTIFICATION_CHANNEL_ID }}"
+            text: >+
+              :x: Scheduled test workflow failed
+              [ ${{ env.WORKFLOW }} ]

label_studio_ml/examples/flair/docker-compose.yml

@@ -14,6 +14,6 @@ services:
       - WORKERS=1
       - THREADS=8
       - LOG_LEVEL=DEBUG
-      - FLAIR_MODEL_NAME=ner
+      - FLAIR_MODEL_NAME=ner-multi
     ports:
       - "9090:9090"

label_studio_ml/examples/flair/model.py

@@ -9,7 +9,7 @@
 
 logger = logging.getLogger(__name__)
 
-FLAIR_MODEL_NAME = os.getenv("FLAIR_MODEL_NAME", "ner")
+FLAIR_MODEL_NAME = os.getenv("FLAIR_MODEL_NAME", "ner-multi")
 logger.info(f"Loading Flair model {FLAIR_MODEL_NAME}")
 _model = Classifier.load(FLAIR_MODEL_NAME)
 

label_studio_ml/examples/flair/requirements.txt

@@ -1,2 +1,3 @@
 scipy==1.10.1
-flair==0.13.1
+torch==2.5.1
+flair==0.15

label_studio_ml/examples/segment_anything_2_image/docker-compose.yml

@@ -16,6 +16,7 @@ services:
               count: 1
               capabilities: [ gpu ]
 
+
     environment:
       # specify these parameters if you want to use basic auth for the model server
       - BASIC_AUTH_USER=

label_studio_ml/examples/yolo/Dockerfile

@@ -11,6 +11,7 @@ RUN --mount=type=cache,target="/var/cache/apt",sharing=locked \
     apt-get -y update \
     && apt-get install -y git \
     && apt-get install -y wget \
+    && apt-get install -y curl \
     && apt-get install -y g++ freeglut3-dev build-essential libx11-dev \
     libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev libfreeimage-dev \
     && apt-get -y install ffmpeg libsm6 libxext6 libffi-dev python3-dev python3-pip gcc