Cropinky commited on
Commit
79a5268
·
1 Parent(s): 94e98b0
Files changed (2) hide show
  1. app.py +1 -1
  2. image_generator.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from image_generator import generate_images
10
 
11
  def image_generation(model, number_of_images=1):
12
  G = MyGenerator.from_pretrained("Cropinky/projected_gan_impressionism")
13
- generate_images(seeds = 10-15, outdir="out")
14
  return f"generating {number_of_images} images from {model}"
15
  if __name__ == "__main__":
16
 
 
10
 
11
  def image_generation(model, number_of_images=1):
12
  G = MyGenerator.from_pretrained("Cropinky/projected_gan_impressionism")
13
+ generate_images()
14
  return f"generating {number_of_images} images from {model}"
15
  if __name__ == "__main__":
16
 
image_generator.py CHANGED
@@ -67,7 +67,7 @@ def make_transform(translate: Tuple[float,float], angle: float):
67
 
68
  @click.command()
69
  @click.option('--network', 'network_pkl', help='Network pickle filename', required=False)
70
- @click.option('--seeds', type=parse_range, help='List of random seeds (e.g., \'0,1,4-6\')', required=True)
71
  @click.option('--trunc', 'truncation_psi', type=float, help='Truncation psi', default=1, show_default=True)
72
  @click.option('--class', 'class_idx', type=int, help='Class label (unconditional if not specified)')
73
  @click.option('--noise-mode', help='Noise mode', type=click.Choice(['const', 'random', 'none']), default='const', show_default=True)
 
67
 
68
  @click.command()
69
  @click.option('--network', 'network_pkl', help='Network pickle filename', required=False)
70
+ @click.option('--seeds', type=parse_range, help='List of random seeds (e.g., \'0,1,4-6\')', default=10-15, required=True)
71
  @click.option('--trunc', 'truncation_psi', type=float, help='Truncation psi', default=1, show_default=True)
72
  @click.option('--class', 'class_idx', type=int, help='Class label (unconditional if not specified)')
73
  @click.option('--noise-mode', help='Noise mode', type=click.Choice(['const', 'random', 'none']), default='const', show_default=True)