Skip to content

Commit

Permalink
Add onnxruntime requirement
Browse files Browse the repository at this point in the history
  • Loading branch information
MalloryWittwer committed Jan 7, 2025
1 parent 03bbcca commit c659182
Show file tree
Hide file tree
Showing 3 changed files with 51 additions and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM imaging-server-kit:3.9
FROM registry.rcp.epfl.ch/imaging-server-kit/imaging-server-kit:3.9

COPY . .

Expand Down
47 changes: 47 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
```
3 changes: 3 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1 +1,4 @@
imaging-server-kit
rembg
onnx
onnxruntime

0 comments on commit c659182

Please sign in to comment.