Spaces:
Sleeping
Sleeping
Commit
•
0c01b4c
1
Parent(s):
bcd203a
Support ONNX models (#2)
Browse files- Support ONNX models (d8bf55471d052a89e1c5683864cf317b291582ec)
- Update Dockerfile (7b0aa23a888ef1654dcdf79871a746e18b7c620b)
Co-authored-by: Justin Chu <[email protected]>
- Dockerfile +1 -1
- 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
|