apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): add time to filenames (for #55)

This commit is contained in:
Sean Sube 2023-01-15 14:22:51 -06:00
parent df6b07194b
commit 16108ae172
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/serve.py
View File

@ -45,6 +45,7 @@ from .image import (
import json import json
import numpy as np import numpy as np
import time
# paths # paths
bundle_path = environ.get('ONNX_WEB_BUNDLE_PATH', bundle_path = environ.get('ONNX_WEB_BUNDLE_PATH',
@ -163,8 +164,10 @@ def serve_bundle_file(filename='index.html'):
def make_output_path(mode: str, seed: int, params: Tuple[Union[str, int, float]]): def make_output_path(mode: str, seed: int, params: Tuple[Union[str, int, float]]):
now = int(time.time())
sha = sha256() sha = sha256()
sha.update(mode.encode('utf-8')) sha.update(mode.encode('utf-8'))
for param in params: for param in params:
if param is None: if param is None:
continue continue
@ -177,7 +180,7 @@ def make_output_path(mode: str, seed: int, params: Tuple[Union[str, int, float]]
else: else:
print('cannot hash param: %s, %s' % (param, type(param))) print('cannot hash param: %s, %s' % (param, type(param)))
output_file = '%s_%s_%s.png' % (mode, seed, sha.hexdigest()) output_file = '%s_%s_%s_%s.png' % (mode, seed, sha.hexdigest(), now)
output_full = safer_join(output_path, output_file) output_full = safer_join(output_path, output_file)
return (output_file, output_full) return (output_file, output_full)