model path arg added, Fixed

kellywujy · kellywujy · commit 1879a0b8e5a4 · 2023-06-29T09:42:21.000-07:00
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -15,6 +15,8 @@ services:
       - OUTPUT_PATH=/outputs/
       - SEND_XDD=False
       - DOI_FILE_PATH=/raw/gdd_api_return.json
+      - MODEL_PATH=/models/logistic_regression_model.joblib
+
     volumes:
       - ./data/article-relevance/outputs:/output
       - ./data/article-relevance/processed/prediction_parquet:/parquet
diff --git a/docker/article-relevance/run-prediction.sh b/docker/article-relevance/run-prediction.sh
@@ -12,5 +12,6 @@ python src/article_relevance/gdd_api_query.py \
 
 python src/article_relevance/relevance_prediction_parquet.py \
   --doi_file_path="$DOI_FILE_PATH" \
+  --model_path="$MODEL_PATH" \
   --output_path="$OUTPUT_PATH" \
   --send_xdd="$SEND_XDD"
