Entrypoint for testing

2023-11-20 18:56:00 -05:00
parent 8f17764f8a
commit c34d79ea49
2 changed files with 29 additions and 0 deletions
@@ -57,6 +57,9 @@ ENV MODEL_NAME=$MODEL_NAME \
    MODEL_BASE_PATH=$MODEL_BASE_PATH \
    HUGGING_FACE_HUB_TOKEN=$HUGGING_FACE_HUB_TOKEN

+# Set the entrypoint
+ENTRYPOINT ["/entrypoint.sh"]
+
 # Run the Python script to download the model
 RUN python -u /download_model.py