Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+
+import torch
 from diffusers import DDPMPipeline, DDIMPipeline, PNDMPipeline
 
 MODEL_ID = "verkaDerkaDerk/tiki-based-128"
@@ -6,17 +8,25 @@ MODEL_ID = "verkaDerkaDerk/tiki-64"
 PIPELINE = DDPMPipeline.from_pretrained(MODEL_ID)
 
 
-def imagineAll():
+def imagineAll(batch_size=1):
+    generator = torch.manual_seed(0)
+    return pipeline(generator=generator, batch_size=batch_size)["sample"]
     return PIPELINE()["sample"]
     
+def imagine4():
+    return imagineAll(4)
+    
 def imagine():
     return imagineAll[0]
 
 def fancy():
     # try some https://huggingface.co/spaces/stabilityai/stable-diffusion/blob/main/app.py trix
-    # for some bizarre reason this results in huggingface.co 'Temporary failure in name resolution'
-    # it works fine in collab...
-    block = gr.Blocks()
+    css = '''
+        .output_image   {height: 40rem !important; width: 100% !important;  } 
+        .object-contain {height: 256px !important; width: 256px !important; }
+        #gallery img    { height: 256px !important; width: 256px !important; }
+    '''
+    block = gr.Blocks(css=css)
     with block:
         gr.HTML('''
             <pre>
@@ -42,9 +52,10 @@ def fancy():
                        
             gallery = gr.Gallery(
                 label="Generated images", show_label=False, elem_id="gallery"
-            ).style(grid=[1], height="256px")
+            ).style(grid=[4], height="256")
+            
+            btn.click(imagine4, inputs=None, outputs=gallery)
             
-            btn.click(imagineAll, inputs=None, outputs=gallery)
             gr.HTML('''
                 <p><img src="https://huggingface.co/spaces/verkaDerkaDerk/tiki-64/resolve/main/tiki-600e.png"></p>