From 74a9ee70020ffa2746c82300c533de3f7e523f22 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 15 Oct 2022 17:25:35 +0300
Subject: fix saving images compatibility with gradio update

---
 modules/ui.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

(limited to 'modules')

diff --git a/modules/ui.py b/modules/ui.py
index 3206113e..b867d40f 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -154,10 +154,7 @@ def save_files(js_data, images, do_make_zip, index):
             writer.writerow(["prompt", "seed", "width", "height", "sampler", "cfgs", "steps", "filename", "negative_prompt"])
 
         for image_index, filedata in enumerate(images, start_index):
-            if filedata.startswith("data:image/png;base64,"):
-                filedata = filedata[len("data:image/png;base64,"):]
-
-            image = Image.open(io.BytesIO(base64.decodebytes(filedata.encode('utf-8'))))
+            image = image_from_url_text(filedata)
 
             is_grid = image_index < p.index_of_first_image
             i = 0 if is_grid else (image_index - p.index_of_first_image)
-- 
cgit v1.2.1