Genzo1010 commited on
Commit
43b9e82
1 Parent(s): df8eef7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -23,4 +23,6 @@ RUN pip install torch torchvision
23
  # Set environment variable to indicate PyTorch usage
24
  ENV USE_TORCH=1
25
 
26
- CMD fastapi run --reload --host=0.0.0.0 --port=7860
 
 
 
23
  # Set environment variable to indicate PyTorch usage
24
  ENV USE_TORCH=1
25
 
26
+ EXPOSE 7860
27
+
28
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]