Update app.py

app.py

@@ -53,6 +53,9 @@ def make_html_table(caption):
     html += '</table>'
     return html
 
+def create_dummy_image(width=512, height=512, alpha=0):
+    return Image.new("RGBA", (width, height), (0, 0, 0, alpha))
+
 def update_history(img, gallery_items):
     tables_html = "".join(
         f'<div style="margin-bottom:12px">{make_html_table(item[1])}</div>'
@@ -97,8 +100,7 @@ def infer(prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type,
     history.insert(0, (img, caption_text))
     progress(1.0, desc="Done!")
 
-#    gallery_items = [(item[0], make_html_table(item[1])) for item in history]
-    gallery_items = [(item, make_html_table(caption)) for item in history]
+    gallery_items = [(img_obj, make_html_table(raw_caption_text)) for img_obj, raw_caption_text in history]
     tables_html = "".join(f'<div style="margin-bottom:12px">{make_html_table(item[1])}</div>' for item in history)
     return img, gallery_items
 
@@ -286,7 +288,11 @@ window.addEventListener('load', () => {
 </script>
             """
         )
-        img_out = gr.Image(interactive=None,value=None)
+        img_out = gr.Image(
+            interactive=None, 
+            value=create_dummy_image(width=512, height=512, alpha=0),
+            label="生成画像" 
+        )
         state   = gr.State([])
         history_gallery = gr.Gallery(
             label="生成履歴",