meeww commited on
Commit
34de9ca
·
1 Parent(s): 1bb879c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,7 +4,7 @@ from image_wgan import ImageWgan
4
  import os
5
  from os.path import exists
6
  from PIL import Image
7
- def init():
8
  generated_samples_folder = "."
9
  discriminator_saved_model = "discriminator64.model"
10
  generator_saved_model = "generator64.model"
@@ -16,7 +16,8 @@ def init():
16
  discriminator_saved_model=discriminator_saved_model if exists(discriminator_saved_model) else None
17
  )
18
  image_wgan.generate(
19
- sample_folder=generated_samples_folder
 
20
  )
21
  crop()
22
 
 
4
  import os
5
  from os.path import exists
6
  from PIL import Image
7
+ def init(seed):
8
  generated_samples_folder = "."
9
  discriminator_saved_model = "discriminator64.model"
10
  generator_saved_model = "generator64.model"
 
16
  discriminator_saved_model=discriminator_saved_model if exists(discriminator_saved_model) else None
17
  )
18
  image_wgan.generate(
19
+ sample_folder=generated_samples_folder,
20
+ seed
21
  )
22
  crop()
23