Commit b46b97fa authored by AUTOMATIC's avatar AUTOMATIC
Browse files

more fixes for gradio update

parent 76f256fe
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -38,7 +38,7 @@ def quote(text):
def image_from_url_text(filedata):
    if type(filedata) == dict and filedata["is_file"]:
        filename = filedata["name"]
        is_in_right_dir = any(Path(temp_dir).resolve() in Path(filename).resolve().parents for temp_dir in shared.demo.temp_dirs)
        is_in_right_dir = any([filename in fileset for fileset in shared.demo.temp_file_sets])
        assert is_in_right_dir, 'trying to open image file outside of allowed directories'

        return Image.open(filename)
+1 −1
Original line number Diff line number Diff line
@@ -45,7 +45,7 @@ def on_tmpdir_changed():

    os.makedirs(shared.opts.temp_dir, exist_ok=True)

    shared.demo.temp_dirs = shared.demo.temp_dirs | {os.path.abspath(shared.opts.temp_dir)}
    shared.demo.temp_file_sets[0] = shared.demo.temp_file_sets[0] | {os.path.abspath(shared.opts.temp_dir)}


def cleanup_tmpdr():