diff --git a/predict.py b/predict.py index 093cf89..e72d0eb 100644 --- a/predict.py +++ b/predict.py @@ -3,7 +3,7 @@ import mlflow import sys import numpy as np -input = sys.argv[1] +#input = sys.argv[1] logged_model = 'mlruns/1/70439eb482b54d56b54b0ecc6f1ca96f/artifacts/s444409' loaded_model = mlflow.pyfunc.load_model(logged_model)