syaikhipin commited on
Commit
001dfad
1 Parent(s): 60bb0aa

Update DockerFile

Browse files
Files changed (1) hide show
  1. DockerFile +1 -0
DockerFile CHANGED
@@ -28,5 +28,6 @@ RUN pip install --no-cache -r requirements.txt \
28
  && pip uninstall torch tensorflow transformers triton -y \
29
  && rm -rf /usr/local/lib/python3.10/site-packages/nvidia*
30
 
 
31
  # Specify the command to run when the Docker container starts
32
  CMD [ "python", "main.py" ]
 
28
  && pip uninstall torch tensorflow transformers triton -y \
29
  && rm -rf /usr/local/lib/python3.10/site-packages/nvidia*
30
 
31
+ EXPOSE 7860
32
  # Specify the command to run when the Docker container starts
33
  CMD [ "python", "main.py" ]