diff --git a/server/visualdl/graph.py b/server/visualdl/graph.py
index 1ad959f656661784c9e861e4103d3ea86731873b..fcb2e8abb5f74b78eef3c32ba08c2500c408fde2 100644
--- a/server/visualdl/graph.py
+++ b/server/visualdl/graph.py
@@ -41,5 +41,9 @@ def load_model(model_pb_path):
     return json.dumps(json_obj, sort_keys=True, indent=4, separators=(',', ': '))
 
 
-json_str = load_model("/Users/qiaolongfei/project/onnx/model/inception_v1/model.pb")
-print(json_str)
+if __name__ == '__main__':
+    import os
+    import sys
+    current_path = os.path.abspath(os.path.dirname(sys.argv[0]))
+    json_str = load_model(current_path + "/mock/inception_v1.pb")
+    print(json_str)