Update app.py

app.py

@@ -36,11 +36,11 @@ def create_collage(uploaded_files, grid_layout):
         col = i % cols
         collage.paste(img, (col * img.width, row * img.height))
 
-    # Save the collage as a PNG file in memory
+    # Save the collage as a PNG file in memory (using BytesIO)
     output_io = io.BytesIO()
     collage.save(output_io, format="PNG")
     output_io.seek(0)  # Reset the pointer to the start of the file
-    return output_io
+    return output_io  # Return the BytesIO object for download
 
 # Gradio Interface
 with gr.Blocks() as app: