add one more layer to model dir

2023-04-11 11:56:56 +08:00
parent 0e6a5b8925
commit ac6400e93a
2 changed files with 2 additions and 2 deletions
--- a/predict.py
+++ b/predict.py
@ -21,7 +21,7 @@ def main():
    print(my_fortune)

    # save model and env to local file system
-    mlflow.pytorch.save_model(model, './model')
+    mlflow.pytorch.save_model(model, './model/fortune_predict_model')


 if __name__ == '__main__':