Update image_wgan.py

image_wgan.py

@@ -9,8 +9,8 @@ from torch.utils.data import DataLoader
 from torchvision.utils import save_image
 
 from image_dataset import ImageDataset
-from wgan.discriminator import Discriminator
-from wgan.generator import Generator
+from discriminator import Discriminator
+from generator import Generator
 
 
 class ImageWgan: