Update app.py

app.py

@@ -9,7 +9,7 @@ from PIL import Image
 # --------------------------
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 latent_dim = 100
-model_path = "generator.pth"  # فایل مدل Generator آموزش دیده
+model_path = "generator.pth"
 
 # --------------------------
 # تعریف Generator
@@ -43,7 +43,7 @@ G.eval()
 def generate_images(seed=42, num_images=4):
     torch.manual_seed(seed)
     z = torch.randn(num_images, latent_dim).to(device)
-    imgs = G(z).detach().cpu().numpy()  # خروجی: (num_images,1,28,28)
+    imgs = G(z).detach().cpu().numpy()
     
     pil_images = []
     for i in range(num_images):
@@ -62,7 +62,7 @@ iface = gr.Interface(
         gr.Slider(0, 10000, value=42, label="Seed"),
         gr.Slider(1, 16, value=4, label="Number of Images")
     ],
-    outputs=gr.Gallery(label="Generated MNIST Images").style(grid=[4,4]),
+    outputs=gr.Gallery(label="Generated MNIST Images", columns=4, type="pil"),
     title="MNIST GAN Generator",
     description="یک مدل GAN برای تولید چند تصویر اعداد دست‌نویس MNIST"
 )