don't resize small images

app.py

@@ -23,8 +23,6 @@ def open_dataset(dataset, set_name, split, batch_size, state, shard = -1):
         data_files = {"train":[f"{path}/{split}-{shard:05d}-of-{shards:05d}.parquet"]}
 
     if DEBUG:
-        ds  = lambda:None
-        ds.n_shards = 1234
         dsi = range(100)
     else:
         ds = load_dataset(
@@ -54,7 +52,9 @@ def get_images(batch_size, state):
     
     for i in tqdm.trange(batch_size, desc=f"Getting images"):
         if DEBUG:
-            image = np.random.randint(0,255,(384,384,3))
+            images.append(np.random.randint(0,255,(384,384,3)))
+            images.append(np.random.randint(0,255,(100,100,3)))
+
             metadata = {"bounds":[[1,1,4,4]], }
         else:
             try:
@@ -96,6 +96,7 @@ if __name__ == "__main__":
         gallery = gr.Gallery(
             label=ev.DATASET,
             interactive=False,
+            object_fit="scale-down",
             columns=5, rows=2, render=False)
 
         with gr.Row():