Update app.py

app.py

@@ -68,11 +68,12 @@ def infer(
         generator=generator,
     ).images[0]
 
-    # Save image as Base64
-    buffered = BytesIO()
-    image.save(buffered, format="PNG")
-    base64_image = base64.b64encode(buffered.getvalue()).decode("utf-8")
-    return image, seed, f"data:image/png;base64,{base64_image}"
+    # # Save image as Base64
+    # buffered = BytesIO()
+    # image.save(buffered, format="PNG")
+    # base64_image = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    # return image, seed, f"data:image/png;base64,{base64_image}"
+    return image, seed
 
 
 examples = ["murasame \(senren\), senren banka",]
@@ -101,7 +102,7 @@ with gr.Blocks() as img:
 
         result = gr.Image(label="Result", show_label=False)
 
-        base64_view = gr.HTML(label="Base64 Image Preview", interactive=True)
+        # base64_view = gr.HTML(label="Base64 Image Preview", interactive=True)
 
         with gr.Accordion("Advanced Settings", open=False):
             negative_prompt = gr.Text(
@@ -173,18 +174,19 @@ with gr.Blocks() as img:
             guidance_scale,
             num_inference_steps,
         ],
-        outputs=[result, seed, base64_view],
+        # outputs=[result, seed, base64_view],
+        outputs=[result, seed],
     )
 
-    # JavaScript logic to dynamically update HTML with Base64
-    js_script = """
-    <script>
-        function updateBase64(html_id, base64_src) {
-            document.getElementById(html_id).innerHTML = `<img src="${base64_src}" alt="Generated Image"/>`;
-        }
-    </script>
-    """
-    gr.HTML(js_script)
+    # # JavaScript logic to dynamically update HTML with Base64
+    # js_script = """
+    # <script>
+    #     function updateBase64(html_id, base64_src) {
+    #         document.getElementById(html_id).innerHTML = `<img src="${base64_src}" alt="Generated Image"/>`;
+    #     }
+    # </script>
+    # """
+    # gr.HTML(js_script)
 
 if __name__ == "__main__":
     img.launch()