jeremyrmanning commited on
Commit
6bea593
·
1 Parent(s): 60c555f

decreased resolution, various minor tweaks

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -38,8 +38,8 @@ def generate_story(choice, input_text):
38
  return generated_text
39
 
40
 
41
- def generate_images(text, width=256, height=256, steps=50, num_images=2,
42
- diversity=3):
43
 
44
  image_bytes = image_gen(text, steps, width, height, num_images, diversity)
45
 
@@ -65,7 +65,7 @@ def generate_interpolation(text, fps=5):
65
  frames.append(f'frame_{i}.png')
66
  g.save(frames[-1])
67
 
68
- vid = list(util.interpolate_recursively_from_files(frames, len(generated_images), interpolator))
69
  mediapy.write_video("out.mp4", vid, fps=fps)
70
  return "out.mp4"
71
 
 
38
  return generated_text
39
 
40
 
41
+ def generate_images(text, width=128, height=128, steps=40, num_images=1,
42
+ diversity=2):
43
 
44
  image_bytes = image_gen(text, steps, width, height, num_images, diversity)
45
 
 
65
  frames.append(f'frame_{i}.png')
66
  g.save(frames[-1])
67
 
68
+ vid = list(util.interpolate_recursively_from_files(frames, np.max([4, len(generated_images)]), interpolator))
69
  mediapy.write_video("out.mp4", vid, fps=fps)
70
  return "out.mp4"
71