Update app.py

app.py

@@ -1,8 +1,7 @@
-# app.py
 import torch
 import torch.nn as nn
 import gradio as gr
-import matplotlib.pyplot as plt
+import numpy as np
 
 # --------------------------
 # تنظیمات
@@ -12,7 +11,7 @@ latent_dim = 100
 model_path = "generator.pth"  # فایل مدل Generator آموزش دیده
 
 # --------------------------
-# تعریف Generator (همان که آموزش داده بودیم)
+# تعریف Generator
 # --------------------------
 class Generator(nn.Module):
     def __init__(self):
@@ -44,7 +43,8 @@ def generate_image(seed=42):
     torch.manual_seed(seed)
     z = torch.randn(1, latent_dim).to(device)
     img = G(z).detach().cpu().squeeze().numpy()
-    img = (img + 1) / 2  # تبدیل مقادیر از [-1,1] به [0,1]
+    img = (img + 1) / 2  # تبدیل [-1,1] به [0,1]
+    img = (img * 255).astype(np.uint8)  # تبدیل به uint8
     return img
 
 # --------------------------
@@ -53,7 +53,7 @@ def generate_image(seed=42):
 iface = gr.Interface(
     fn=generate_image,
     inputs=gr.Slider(0, 10000, value=42, label="Seed"),
-    outputs=gr.Image(shape=(28,28)),
+    outputs=gr.Image(type="numpy"),  # مهم: از type="numpy" استفاده کنید
     title="MNIST GAN Generator",
     description="یک مدل GAN ساده برای تولید تصاویر اعداد دست‌نویس MNIST"
 )