diff --git a/utils/store.py b/utils/store.py index c48ff2c..50e3a6f 100644 --- a/utils/store.py +++ b/utils/store.py @@ -298,7 +298,7 @@ def save_html( else: raise ValueError(f"Invalid type: {type(html_data)}") if static: - static_path = os.path.join(model_path, label + ".png") + static_path = os.path.join(model_path, label + ".pdf") imgkit.from_file(path, static_path) return path