Added Dockerfile
This commit is contained in:
14
README.md
14
README.md
@@ -17,3 +17,17 @@ You can experience their great online services [here](https://cleanup.pictures/)
|
||||
- Install dependencies:`cd lama_cleaner/app/ && yarn`
|
||||
- Start development server: `yarn dev`
|
||||
- Build: `yarn build`
|
||||
|
||||
## Docker (cpu)
|
||||
```
|
||||
docker build -f Dockerfile -t lamacleaner .
|
||||
docker run -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd):/app --rm lamacleaner python3 main.py --device=cpu --port=8080
|
||||
```
|
||||
|
||||
## Docker (gpu)
|
||||
```
|
||||
docker build -f Dockerfile -t lamacleaner .
|
||||
docker run --gpus all -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd):/app --rm lamacleaner python3 main.py --device=gpu --port=8080
|
||||
```
|
||||
|
||||
Then open [http://localhost:8080](http://localhost:8080)
|
||||
Reference in New Issue
Block a user