Refresh fix

This commit is contained in:
Sakura-Luna 2023-05-04 15:55:57 +08:00
parent 91a15dca80
commit 8bc4a3a2a8

View File

@ -35,7 +35,8 @@ def check_tmp_file(gradio, filename):
def save_pil_to_file(pil_image, dir=None): def save_pil_to_file(pil_image, dir=None):
already_saved_as = getattr(pil_image, 'already_saved_as', None) already_saved_as = getattr(pil_image, 'already_saved_as', None)
if already_saved_as and os.path.isfile(already_saved_as): if already_saved_as and os.path.isfile(already_saved_as):
register_tmp_file(shared.demo, f'{already_saved_as}?{int(time())}') already_saved_as += f'?{int(time())}'
register_tmp_file(shared.demo, already_saved_as)
file_obj = Savedfile(already_saved_as) file_obj = Savedfile(already_saved_as)
return file_obj return file_obj