model_path from doc opt

kellywujy · kellywujy · commit cf2e23b1f0bc · 2023-06-29T09:52:15.000-07:00
diff --git a/src/article_relevance/relevance_prediction_parquet.py b/src/article_relevance/relevance_prediction_parquet.py
@@ -425,14 +425,16 @@ def main():
     output_path = opt['--output_path']
     send_xdd = opt['--send_xdd']
     
-    # /models directory is a mounted volume, containing the model object
-    models = os.listdir("/models")
-    models = [f for f in models if f.endswith(".joblib")]
+    # # /models directory is a mounted volume, containing the model object
+    # models = os.listdir("/models")
+    # models = [f for f in models if f.endswith(".joblib")]
     
-    if models:
-        model_path = os.path.join("/models", models[0])
-    else:
-        model_path = ""
+    # if models:
+    #     model_path = os.path.join("/models", models[0])
+    # else:
+    #     model_path = ""
+    
+    model_path = opt['--model_path']
 
     metadata_df = crossref_extract(doi_list_file_path)
 
