Add app.py

app.py

@@ -4,8 +4,8 @@ import jax
 import jax.numpy as jnp
 
 # Dummy simple function to simulate diffusion sampling
-def generate_image(seed):
-    key = jax.random.PRNGKey(seed)
+def generate_image():
+    key = jax.random.PRNGKey(np.random.randint(0, 10000))
     img = jax.random.normal(key, (64, 64, 3))
     img = (img - img.min()) / (img.max() - img.min())  # Normalize to [0, 1]
     return np.array(img)
@@ -13,10 +13,12 @@ def generate_image(seed):
 # Gradio Interface
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Number(label="Random Seed"),
+    inputs=[],
     outputs=gr.Image(shape=(64, 64)),
     title="JAX Diffusion Demo",
-    description="A simple diffusion-style random image generator using JAX.",
+    description="Click 'Generate' to sample a random diffusion image!",
+    allow_flagging="never",
+    live=False
 )
 
 if __name__ == "__main__":