Update score.py

Azure · Sep 1, 2022 · e0c073c · e0c073c
1 parent 5ec5d8a
commit e0c073c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/classical/python-sdk/data-science/src/score.py b/classical/python-sdk/data-science/src/score.py
@@ -22,7 +22,10 @@ def init():
 
     model_path = Model.get_model_path(args.model_name)
     print(f"Model path: {model_path}")
-    model = joblib.load(os.path.join(model_path, 'model.pkl'))
+    if 'model.pkl' in model_path:
+        model = joblib.load(model_path)
+    else:
+        model = joblib.load(os.path.join(model_path, 'model.pkl'))
 
         # load the explainer
     explainer_path = os.path.join(Model.get_model_path(args.model_name), "explainer")
@@ -56,4 +59,4 @@ def run(file_list):
         result = pd.concat([sno, proba], axis=1)
         results = results.append(result)
         print(f"Batch scored: {filename}")
-    return results
+    return results