Add app.py

app.py

@@ -14,13 +14,11 @@ def generate_image():
 iface = gr.Interface(
     fn=generate_image,
     inputs=[],
-    outputs=gr.Image(shape=(64, 64)),
+    outputs=gr.Image(type="numpy", label="Generated Image"),
     title="JAX Diffusion Demo",
     description="Click 'Generate' to sample a random diffusion image!",
-    allow_flagging="never",
-    live=False
+    allow_flagging="never"
 )
 
 if __name__ == "__main__":
     iface.launch()
-