replace bash script to docker-compose to build server

This commit is contained in:
2023-03-07 16:33:45 +08:00
parent ec19042d0d
commit 1f86146b12
7 changed files with 28 additions and 16 deletions

View File

@ -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')