diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-08-21 17:36:17 -0400 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-08-21 17:36:17 -0400 |
| commit | 299b8096bc19c8a619fbeef735cb90544350243c (patch) | |
| tree | cbc35adf9f5b6859271934aa8e15ae5889b3781e /modules/ui_tempdir.py | |
| parent | 953c3eab7b3b952f7e96d728413a531d7fb521a2 (diff) | |
Make Gradio temp directory if it doesn't exist
Gradio normally creates the temp directory in `pil_to_temp_file()` (https://github.com/gradio-app/gradio/blob/861d752a83da0f95e9f79173069b69eababeed39/gradio/components/base.py#L313) but since the Gradio implementation of `pil_to_temp_file()` is replaced with `save_pil_to_file()`, the Gradio temp directory should also be created by `save_pil_to_file()` when necessary.
Diffstat (limited to 'modules/ui_tempdir.py')
| -rw-r--r-- | modules/ui_tempdir.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/ui_tempdir.py b/modules/ui_tempdir.py index 506017e5..85015db5 100644 --- a/modules/ui_tempdir.py +++ b/modules/ui_tempdir.py @@ -44,6 +44,8 @@ def save_pil_to_file(self, pil_image, dir=None, format="png"): if shared.opts.temp_dir != "":
dir = shared.opts.temp_dir
+ else:
+ os.makedirs(dir, exist_ok=True)
use_metadata = False
metadata = PngImagePlugin.PngInfo()
|