Add application and model

app.py

@@ -54,13 +54,13 @@ def inference_interface(latent_dim=latent_dim, device=device):
         upscaled_images = [img.resize((256, 256), resample=Image.LANCZOS) for img in images]
         return upscaled_images
 
-    demo = gr.Interface(
-        fn=generate,
-        inputs=gr.Slider(minimum=1, maximum=9, step=1, value=1, label="Number of Images"),
-        outputs=gr.Gallery(label="Generated Artwork").style(grid=[3], height="auto"),
-        title="Art Generation with GAN",
-        description="Generate artwork using a trained GAN model."
-    )
+    demo = gr.Interface(fn=generate,
+      inputs=gr.Slider(minimum=1, maximum=9, step=1, value=1, label="Number of Images"),
+      outputs=gr.Gallery(label="Generated Artwork", columns=3, height="auto"),
+      title="Art Generation with GAN",
+      description="Generate artwork using a trained GAN model."
+      )
+
     return demo
 
 # The key part: launch the Gradio interface when app.py is run