Update Dockerfile

justinmerrell · justinmerrell · commit 9f24389daaf8 · 2023-05-10T13:09:22.000-04:00
diff --git a/Dockerfile b/Dockerfile
@@ -26,4 +26,4 @@ RUN rm /model_fetcher.py
 # Add src files (Worker Template)
 ADD src .
 
-CMD [ "python", "-u", "/rp_handler.py", "--model_tag=${MODEL_TAG}" ]
+CMD python -u /rp_handler.py --model_tag=${MODEL_TAG}
