Update image_wgan.py

image_wgan.py

@@ -107,7 +107,7 @@ class ImageWgan:
     def generate(
         self,
         sample_folder: str = 'samples',
-        seed: str = 'seed'
+        seed: int = 'seed'
     ):
         if not exists(sample_folder):
             mkdir(sample_folder)