Update app.py

app.py

@@ -74,11 +74,6 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     # Return image, seed, and the used prompts
     return image, seed, f"Prompt used: {final_prompt}\nNegative prompt used: {full_negative_prompt}"
 
-def download_image(image):
-    """
-    Prepares the image for download as a PNG.
-    """
-    return image
 
 css = """
 #col-container {
@@ -136,10 +131,7 @@ with gr.Blocks(css=css) as demo:
             gr.Markdown("""# Rainbow Media X""")
 
             # Display result image at the top
-            result = gr.Image(label="Result", show_label=False, elem_id="result", type="pil")
-
-            # Add a download button
-            download_button = gr.Button("Download as PNG", elem_id="download-button")
+            result = gr.Image(label="Result", show_label=False, elem_id="result")
 
             # Add a textbox to display the prompts used for generation
             prompt_info = gr.Textbox(label="Prompts Used", lines=3, interactive=False, elem_id="prompt-info")
@@ -246,10 +238,4 @@ with gr.Blocks(css=css) as demo:
             outputs=[result, seed, prompt_info]
         )
 
-        download_button.click(
-            download_image,
-            inputs=result,
-            outputs=gr.File(label="Download as PNG")
-        )
-
-demo.queue().launch()
+demo.queue().launch()