log_unsupported_model.py

@@ -64,7 +64,7 @@ def pull_model(server_uri: str,
     print(f'input data: {fake_data}, predictions: {output}')
 
     # save it to local file system
-    mlflow.pyfunc.save_model(path=f'./model/{exp_name}', python_model=model)
+    mlflow.pyfunc.save_model(path='./model', python_model=model)
 
 
 if __name__ == '__main__':

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/fortune_predict_model')
+    mlflow.pytorch.save_model(model, './model')
 
 
 if __name__ == '__main__':