add support for running with gradio 3.9 installed

This commit is contained in:
AUTOMATIC
2023-01-03 14:18:48 +03:00
parent fda1ed1843
commit c0ee148870
2 changed files with 23 additions and 4 deletions

View File

@@ -7,7 +7,7 @@ from pathlib import Path
import gradio as gr
from modules.shared import script_path
from modules import shared
from modules import shared, ui_tempdir
import tempfile
from PIL import Image
@@ -39,7 +39,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([filename in fileset for fileset in shared.demo.temp_file_sets])
is_in_right_dir = ui_tempdir.check_tmp_file(shared.demo, filename)
assert is_in_right_dir, 'trying to open image file outside of allowed directories'
return Image.open(filename)