Update app.py

app.py

@@ -8,16 +8,16 @@ MODEL_ID = "verkaDerkaDerk/tiki-64"
 PIPELINE = DDPMPipeline.from_pretrained(MODEL_ID)
 
 
-def imagineAll(batch_size=1):
-    generator = torch.manual_seed(0)
-    return pipeline(generator=generator, batch_size=batch_size)["sample"]
+def tiki(batch_size=1,seed=0):
+    generator = torch.manual_seed(seed)
+    return PIPELINE(generator=generator, batch_size=batch_size)["sample"]
     return PIPELINE()["sample"]
     
 def imagine4():
-    return imagineAll(4)
+    return tiki(batch_size=4)
     
 def imagine():
-    return imagineAll[0]
+    return tiki()[0]
 
 def fancy():
     # try some https://huggingface.co/spaces/stabilityai/stable-diffusion/blob/main/app.py trix
@@ -54,8 +54,14 @@ def fancy():
                 label="Generated images", show_label=False, elem_id="gallery"
             ).style(grid=[4], height="256")
             
-            btn.click(imagine4, inputs=None, outputs=gallery)
+            #btn.click(imagine4, inputs=None, outputs=gallery)
             
+            with gr.Row(elem_id="tiki-options"):
+                batch_size = gr.Slider(label="batch_size", minimum=1, maximum=4, value=4, step=1)
+                seed = gr.Slider(label="seed", minimum=0, maximum=2022, value=0, step=1)
+
+            btn.click(tiki, inputs=[batch_size,seed], outputs=gallery)
+ 
             gr.HTML('''
                 <p><img src="https://huggingface.co/spaces/verkaDerkaDerk/tiki-64/resolve/main/tiki-600e.png"></p>
 
@@ -75,7 +81,6 @@ def plain():
             - did this change make it to the front-end?
     '''
     article = gr.HTML('''
-        <p><img src="blob/main/tiki-600e.png"></p>
         <p>Trained with <a href="https://github.com/huggingface/diffusers">diffusers</a>.</p>
     ''')