diff options
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/extras.py | 7 | ||||
| -rw-r--r-- | modules/ui.py | 4 |
2 files changed, 6 insertions, 5 deletions
diff --git a/modules/extras.py b/modules/extras.py index 38d6ec48..64b4f2b6 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -97,7 +97,7 @@ def run_extras(image, image_folder, gfpgan_visibility, codeformer_visibility, co return outputs, plaintext_to_html(info), ''
-def run_pnginfo(image):
+def run_image_info(image):
items = image.info
if "exif" in image.info:
@@ -111,8 +111,9 @@ def run_pnginfo(image): items['exif comment'] = exif_comment
- for field in ['jfif', 'jfif_version', 'jfif_unit', 'jfif_density', 'dpi', 'exif']:
- del items[field]
+ for field in ['jfif', 'jfif_version', 'jfif_unit', 'jfif_density', 'dpi', 'exif',
+ 'loop', 'background', 'timestamp', 'duration']:
+ items.pop(field, None)
info = ''
diff --git a/modules/ui.py b/modules/ui.py index 738ac945..0899490f 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -295,7 +295,7 @@ def create_toprow(is_img2img): return prompt, roll, prompt_style, negative_prompt, prompt_style2, submit, interrogate, prompt_style_apply, save_style, check_progress
-def create_ui(txt2img, img2img, run_extras, run_pnginfo):
+def create_ui(txt2img, img2img, run_extras, run_image_info):
with gr.Blocks(analytics_enabled=False) as txt2img_interface:
txt2img_prompt, roll, txt2img_prompt_style, txt2img_negative_prompt, txt2img_prompt_style2, submit, _, txt2img_prompt_style_apply, txt2img_save_style, check_progress = create_toprow(is_img2img=False)
@@ -697,7 +697,7 @@ def create_ui(txt2img, img2img, run_extras, run_pnginfo): submit.click(**extras_args)
pnginfo_interface = gr.Interface(
- wrap_gradio_call(run_pnginfo),
+ wrap_gradio_call(run_image_info),
inputs=[
gr.Image(label="Source", source="upload", interactive=True, type="pil"),
],
|