Update app.py

app.py

@@ -3,6 +3,8 @@ from diffusers import DDPMPipeline, DDIMPipeline, PNDMPipeline
 
 MODEL_ID = "verkaDerkaDerk/tiki-based-128"
 PIPELINE = DDPMPipeline.from_pretrained(MODEL_ID)
+PIPELINE = PIPELINE.to("cuda")
+
 
 #def imagine(name):
 def imagine():
@@ -10,10 +12,31 @@ def imagine():
     return image
 
 def main():
-    inputs = "text"
-    inputs = None
-    outputs = "pil"
-    gr.Interface(fn=imagine, inputs=inputs, outputs=outputs).launch()
+    if True:
+        # try some https://huggingface.co/spaces/stabilityai/stable-diffusion/blob/main/app.py trix
+        block = gr.Blocks()
+        with block:
+            gr.HTML('''
+                maybe this will work?
+                
+            ''')
+            with gr.Group():
+                with gr.Box():
+                    with gr.Row():
+                        btn = gr.Button("Generate image")
+                        btn.click(infer, inputs=[text, samples, steps, scale, seed], outputs=gallery)
+                gallery = gr.Gallery(
+                    label="Generated images", show_label=False, elem_id="gallery"
+                ).style(grid=[1], height="auto")
+                btn.click(imagine, inputs=None, outputs=gallery)
+                gr.HTML('''
+                    Then a nice little footer?
+                '''')
+    else:
+        inputs = "text"
+        inputs = None
+        outputs = "pil"
+        gr.Interface(fn=imagine, inputs=inputs, outputs=outputs).launch()
 
         
 main()