Update app.py

app.py

@@ -62,11 +62,7 @@ def generate_palette_image(colors):
         img = Image.new('RGB', (color_width, img_height), color)
         palette_img.paste(img, (i * color_width, 0))
     
-    # Save the image to a file in memory
-    buf = BytesIO()
-    palette_img.save(buf, format='PNG')
-    buf.seek(0)
-    return buf
+    return palette_img  # Return the PIL image directly
 
 # Function to generate the CSS code for the color palette
 def generate_css_code(colors):
@@ -114,7 +110,7 @@ with gr.Blocks() as interface:
         
         with gr.Column():
             palette_output = gr.HTML(label="Generated Color Palette")
-            palette_image_output = gr.Image(label="Downloadable Palette Image")
+            palette_image_output = gr.Image(label="Downloadable Palette Image", type="pil")  # Output PIL image
             css_code_output = gr.Textbox(label="Generated CSS Code", lines=6)  # Use Textbox to display CSS code
     
     submit_btn.click(gradio_interface, inputs=[image_input], outputs=[