Merge branch 'dev' into release_candidate

1 files changed, 5 insertions, 2 deletions

diff --git a/modules/ui_tempdir.py b/modules/ui_tempdir.py
index fb75137e..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()

@@ -57,8 +59,9 @@ def save_pil_to_file(self, pil_image, dir=None, format="png"):
     return file_obj.name

 

 

-# override save to file function so that it also writes PNG info

-gradio.components.IOComponent.pil_to_temp_file = save_pil_to_file

+def install_ui_tempdir_override():

+    """override save to file function so that it also writes PNG info"""

+    gradio.components.IOComponent.pil_to_temp_file = save_pil_to_file

 

 

 def on_tmpdir_changed():