updates to interface

app.py

@@ -1,20 +1,32 @@
 import gradio as gr
 from diffusers import PNDMPipeline, PNDMScheduler
 
-
-scheduler = PNDMScheduler(num_train_timesteps=1100)
-pipeline = PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
-
-
-def generate_img():
+def update():
     try:
         pipeline.to("cuda")
     except:
         pass
     images= pipeline().images
+    print(images)
     im = images[0]
     return im
 
-iface = gr.Interface(fn=generate_img, inputs=[], outputs="image")
+scheduler = PNDMScheduler(num_train_timesteps=1100)
+pipeline = PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
+
+css = """
+        .container {
+            max-width: 512px;
+            max-height: 512px;
+            margin: auto;
+            padding-top: 1.5rem;
+        }
+    """
+block = gr.Blocks(css=css) 
 
-iface.launch()
+with block as demo:
+    with gr.Row():
+       out = gr.Image()
+    btn = gr.Button("Generate")
+    btn.click(update, inputs=None, outputs=out)
+demo.launch()