diff --git a/visualdl/server/visualDL b/visualdl/server/visualDL index 8487c8adc664e3e22171539934bb19b0fb094141..2ee9d82ad2483056bdee03f183f346f39426ce32 100644 --- a/visualdl/server/visualDL +++ b/visualdl/server/visualDL @@ -84,8 +84,8 @@ mock_data_path = os.path.join(SERVER_DIR, "./mock_data/") log_reader = LogReader(args.logdir) -graph_image_path = None - +# mannully put graph's image on this path also works. +graph_image_path = os.path.join(args.logdir, 'graph.jpg') # return data # status, msg, data @@ -203,7 +203,7 @@ def histogram(): @app.route('/data/plugin/graphs/graph') def graph(): # TODO(ChunweiYan) need to add a config for whether have graph. - if graph_image_path is None: + if graph_image_path is None or not os.path.isfile(graph_image_path): data = {'url': ''} else: data = {'url': '/graphs/image'}