apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

temporarily remove pytorch from API images

This commit is contained in:
Sean Sube 2023-01-06 13:31:18 -06:00
parent 16b7bded25
commit a7a44560db
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/Containerfile.alpine
View File

@ -7,6 +7,6 @@ COPY serve.py /onnx-web/serve.py
RUN pip install -r requirements.txt
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
# RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
ENTRYPOINT [ "flask", "--app=serve", "run", "--host=0.0.0.0" ]

2
api/Containerfile.buster
View File

@ -7,6 +7,6 @@ COPY serve.py /onnx-web/serve.py
RUN pip install -r requirements.txt
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
# RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
ENTRYPOINT [ "flask", "--app=serve", "run", "--host=0.0.0.0" ]