replace bash script to docker-compose to build server
This commit is contained in:
@ -10,7 +10,7 @@ import mlflow
|
||||
if __name__ == '__main__':
|
||||
|
||||
# set MLflow server
|
||||
mlflow.set_tracking_uri('http://127.0.0.1:5000')
|
||||
mlflow.set_tracking_uri('http://127.0.0.1:5001')
|
||||
|
||||
# load production model
|
||||
model = mlflow.pytorch.load_model('models:/fortune_predict_model/production')
|
||||
|
Reference in New Issue
Block a user