Update app.py

app.py

@@ -235,13 +235,28 @@ def update_history(new_image, history):
     return history
 
 
-# --- CSS and Title (unchanged) ---
-# CSS and title (unchanged)
+# --- CSS and Gradio UI ---
 css = """
 h1 {
   text-align: center;
   display: block;
 }
+/* Make gallery items smaller */
+.gradio-container .gallery {
+    grid-template-columns: repeat(auto-fill, minmax(120px, 1fr)) !important; /* Adjust 120px as needed */
+    gap: 10px !important;
+}
+.gradio-container .gallery > .gallery-item {
+    height: auto !important; /* Allow height to adjust based on content */
+    max-height: 150px; /* Set a max height if needed */
+    overflow: hidden; /* Hide overflow */
+}
+.gradio-container .gallery > .gallery-item img {
+    object-fit: contain !important; /* Fit image within bounds */
+    height: 100% !important;
+    width: 100% !important;
+}
+
 """
 
 title = """<h1 align="center">Diffusers Image Outpaint Lightning</h1>