Update app.py

Fix display error that prevented startup

===== Application Startup at 2024-12-23 17:29:12 =====

Traceback (most recent call last):
File "/home/user/app/app.py", line 143, in <module>
display_ui()
File "/home/user/app/app.py", line 122, in display_ui
cool_images_gallery.style(grid=DISPLAY_N_COOL)
AttributeError: 'Gallery' object has no attribute 'style'. Did you mean: 'scale'?

app.py

@@ -85,12 +85,12 @@ def labeling_ui():
 
 # --- Display UI ---
 def display_ui():
-    
+
     def get_random_cool_images(n):
         cool_samples = [d for d in labeled_data if d["label"] == "cool"]
         if len(cool_samples) < n:
           return [Image.frombytes("RGB", (384,384), s["image"]) for s in cool_samples]
-        
+
         selected_cool = random.sample(cool_samples, n)
         return [Image.frombytes("RGB", (384,384), s["image"]) for s in selected_cool]
 
@@ -109,7 +109,7 @@ def display_ui():
         new_image, new_metadata = get_new_unlabeled_image()
         if new_image is None:
             return "No more samples", gr.Image.update(value=None), gr.Gallery.update(value=[])
-        
+
         cool_images = get_random_cool_images(DISPLAY_N_COOL)
         return "", new_image, cool_images
 
@@ -118,8 +118,11 @@ def display_ui():
         metadata_display = gr.Textbox(label="Metadata (Bounds)")
 
     with gr.Row():
-        cool_images_gallery = gr.Gallery(label="Cool Examples", value=[])
-        cool_images_gallery.style(grid=DISPLAY_N_COOL)
+        cool_images_gallery = gr.Gallery(
+            label="Cool Examples",
+            value=[],
+            columns=DISPLAY_N_COOL  # Set grid layout here
+        )
 
     with gr.Row():
         refresh_button = gr.Button("Refresh")