1aurent justinchums commited on
Commit
0c01b4c
1 Parent(s): bcd203a

Support ONNX models (#2)

Browse files

- Support ONNX models (d8bf55471d052a89e1c5683864cf317b291582ec)
- Update Dockerfile (7b0aa23a888ef1654dcdf79871a746e18b7c620b)


Co-authored-by: Justin Chu <justinchums@users.noreply.huggingface.co>

Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +1 -0
Dockerfile CHANGED
@@ -15,4 +15,4 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
15
 
16
  COPY --chown=user . /app
17
 
18
- CMD ["model-explorer", "--host=0.0.0.0", "--port=7860", "--no_open_in_browser"]
 
15
 
16
  COPY --chown=user . /app
17
 
18
+ CMD ["model-explorer", "--extensions=model_explorer_onnx", "--host=0.0.0.0", "--port=7860", "--no_open_in_browser"]
requirements.txt CHANGED
@@ -1 +1,2 @@
1
  ai-edge-model-explorer
 
 
1
  ai-edge-model-explorer
2
+ model-explorer-onnx