Add app.py

app.py

@@ -2,7 +2,6 @@ import jax
 import jax.numpy as jnp
 import gradio as gr
 import numpy as np
-import matplotlib.pyplot as plt
 
 # Dummy diffusion sampling function
 def generate_image(seed):
@@ -19,7 +18,7 @@ with gr.Blocks() as demo:
         seed_slider = gr.Slider(minimum=0, maximum=10000, step=1, value=42, label="Seed")
         generate_button = gr.Button("Generate Image")
     
-    output_image = gr.Image(type="numpy", label="Generated Image", shape=(64, 64))
+    output_image = gr.Image(type="numpy", label="Generated Image")  # 🚫 Removed shape=(64, 64)
 
     generate_button.click(
         fn=generate_image,