From c65918226cfa2c293c62845cb4878de84c626ad8 Mon Sep 17 00:00:00 2001 From: Mallory Wittwer Date: Tue, 7 Jan 2025 09:37:09 +0100 Subject: [PATCH] Add onnxruntime requirement --- Dockerfile | 2 +- README.md | 47 +++++++++++++++++++++++++++++++++++++++++++++++ requirements.txt | 3 +++ 3 files changed, 51 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index ae41b75..605e3fa 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM imaging-server-kit:3.9 +FROM registry.rcp.epfl.ch/imaging-server-kit/imaging-server-kit:3.9 COPY . . diff --git a/README.md b/README.md index 1186679..bdd29f1 100644 --- a/README.md +++ b/README.md @@ -4,3 +4,50 @@ Implementation of a web API server for [rembg](https://github.com/danielgatis/rembg). Author: EPFL Center for Imaging + +## Installation with `pip` + +Install dependencies: + +``` +pip install -r requirements.txt +``` + +Run the server: + +``` +uvicorn main:app --host 0.0.0.0 --port 8000 +``` + +Running tests: + +``` +pytest +``` + +## Installation with `docker` + +Build the docker image: + +``` +docker build -t serverkit-rembg . +``` + +Run the server in a container: + +``` +docker run -it --rm -p 8000:8000 serverkit-rembg:latest +``` + +Running tests: + +``` +docker run --rm serverkit-rembg:latest pytest +``` + +Pushing the image to `registry.rcp.epfl.ch`: + +``` +docker tag serverkit-rembg registry.rcp.epfl.ch/imaging-server-kit/serverkit-rembg +docker push registry.rcp.epfl.ch/imaging-server-kit/serverkit-rembg +``` \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index ed2ce86..3502d97 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,4 @@ +imaging-server-kit rembg +onnx +onnxruntime \ No newline at end of file