add docker container

1123
Feng 4 years ago
parent dc5ec4b18d
commit f92e5253a4

@ -1,3 +1,6 @@
#!/usr/bin/env python3.8
from flask import Flask, render_template, request
from onnx_modifier import onnxModifier
app = Flask(__name__)
@ -30,4 +33,5 @@ def modify_and_download_model():
return 'OK', 200
if __name__ == '__main__':
app.run()
app.debug = True
app.run(host="0.0.0.0")

@ -37,8 +37,7 @@ Clone the repo and install the required Python packages by
git clone git@github.com:ZhangGe6/onnx-modifier.git
cd onnx-modifier
pip install onnx
pip install flask
pip install onnx onnxruntime flask
```
Then run
@ -58,6 +57,24 @@ Click the url in the output info generated by flask (`http://127.0.0.1:5000/` fo
Click `Open Model...` to upload the ONNX model to edit. The model will be parsed and shown on the page.
## launch from a docker container
We create a docker container like this:
```bash
docker build --file Dockerfile . -t onnx-modifier
```
After building the container, we run onnx-modifier from it by mapping docker port 5000 to host port 5000
```bash
docker run -d -t --name onnx-modifier -p 5000:5000 onnx-modifier
```
Then we have access to onnx-modifer from URL <http://127.0.0.1:5000>.
# Usage
Graph-level-operation elements are placed on the left-top of the page. Currently, there are four buttons: `Refresh`, `Reset`, `Download` and `Add node`. They can do:
- `Refresh`: Refresh the model graph to preview editing effects.

Loading…
Cancel
Save