Merge pull request #6 from semi-technologies/switch-to-github-actions

Switch to github actions and update dependencies to newest versions

Author: antas-marcin

.github/CODEOWNERS

@@ -0,0 +1,3 @@
+# Ci related folders
+/.github/ @semi-technologies/core
+/cicd/ @semi-technologies/core

.github/workflows/main.yaml

@@ -0,0 +1,68 @@
+name: Main
+
+on:
+  push:
+    branches:
+      - main
+    tags:
+      - '**'
+    paths-ignore:
+      - README.md
+      - LICENSE
+  pull_request:
+
+jobs:
+  build-image:
+    name: Build image
+    runs-on: ubuntu-latest
+    env:
+      LOCAL_REPO: img2vec-pytorch
+      REMOTE_REPO: semitechnologies/img2vec-pytorch
+      MODEL_NAME: resnet50
+    steps:
+      - uses: actions/checkout@v3
+      - uses: actions/setup-python@v4
+        with:
+          python-version: "3.10"
+      - name: Login to Docker Hub
+        if: ${{ !github.event.pull_request.head.repo.fork }}
+        uses: docker/login-action@v2
+        with:
+          username: ${{secrets.DOCKER_USERNAME}}
+          password: ${{secrets.DOCKER_PASSWORD}}
+      - name: Build and test
+        run: |
+          cicd/build.sh || exit 1
+          cicd/test.sh || exit 1
+      - name: Deploy
+        if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags')
+        env:
+          DOCKER_USERNAME: ${{secrets.DOCKER_USERNAME}}
+          DOCKER_PASSWORD: ${{secrets.DOCKER_PASSWORD}}
+        run: |
+          export GIT_TAG=${GITHUB_REF##*/}
+          cicd/docker_push.sh || exit 1
+  build-custom-image:
+    name: Build custom image
+    if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags')
+    runs-on: ubuntu-latest
+    env:
+      LOCAL_REPO: img2vec-pytorch
+      REMOTE_REPO: semitechnologies/img2vec-pytorch
+    steps:
+      - uses: actions/checkout@v3
+      - uses: actions/setup-python@v4
+        with:
+          python-version: "3.10"
+      - name: Login to Docker Hub
+        uses: docker/login-action@v2
+        with:
+          username: ${{secrets.DOCKER_USERNAME}}
+          password: ${{secrets.DOCKER_PASSWORD}}
+      - name: Deploy
+        env:
+          DOCKER_USERNAME: ${{secrets.DOCKER_USERNAME}}
+          DOCKER_PASSWORD: ${{secrets.DOCKER_PASSWORD}}
+        run: |
+          export GIT_TAG=${GITHUB_REF##*/}
+          cicd/build_custom_base.sh || exit 1

.travis.yml

@@ -1,9 +1,10 @@
-FROM python:3.8-slim-buster
+FROM python:3.10-slim-bullseye
 
 WORKDIR /app
 
 COPY requirements.txt .
-RUN pip3 install -r requirements.txt
+RUN apt-get update && apt-get upgrade && apt-get -y install curl build-essential && curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y && pip3 install -r requirements.txt && apt remove -y curl build-essential && apt -y autoremove
+ENV PATH="$PATH:/root/.cargo/bin"
 
 ARG MODEL_NAME
 COPY download_model.py .

Dockerfile

@@ -36,6 +36,7 @@ def live_and_ready(response: Response):
 
 
 @app.post("/vectors")
+@app.post("/vectors/")
 def read_item(item: VectorImagePayload, response: Response):
 	try:
 		vector = imgVec.vectorize(item.id, item.image)

app.py

@@ -1,5 +1,5 @@
-fastapi==0.85.1
-torch==1.13.0
-torchvision==0.14.0
-uvicorn==0.19.0
-Pillow==9.2.0
+fastapi==0.89.1
+torch==1.13.1
+torchvision==0.14.1
+uvicorn==0.20.0
+Pillow==9.4.0