ClemSummer commited on
Commit
629276e
·
1 Parent(s): 5dafda7

Add CMD to run main.py

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -23,4 +23,4 @@ RUN mkdir -p /models/clip && \
23
  RUN python3 -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('bert-base-uncased').save_pretrained('/models/bert-tokenizer')"
24
  RUN python3 -c "from transformers import CLIPProcessor; CLIPProcessor.from_pretrained('openai/clip-vit-base-patch32').save_pretrained('/models/clip')"
25
  EXPOSE 8000
26
- CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
 
23
  RUN python3 -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('bert-base-uncased').save_pretrained('/models/bert-tokenizer')"
24
  RUN python3 -c "from transformers import CLIPProcessor; CLIPProcessor.from_pretrained('openai/clip-vit-base-patch32').save_pretrained('/models/clip')"
25
  EXPOSE 8000
26
+ CMD ["python", "main.py"]