Ahsen Khaliq commited on
Commit
eb99475
1 Parent(s): 3aa1627

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -108,7 +108,7 @@ w_stds = G.mapping(zs, None).std(0)
108
  def inference(text):
109
  all_frames = []
110
  target = clip_model.embed_text(text)
111
- steps = 30
112
  seed = 2
113
  tf = Compose([
114
  Resize(224),
@@ -152,7 +152,7 @@ def inference(text):
152
  all_frames.append(pil_image)
153
  #os.makedirs(f'samples/{timestring}', exist_ok=True)
154
  #pil_image.save(f'samples/{timestring}/{i:04}.jpg')
155
- writer = imageio.get_writer('test.mp4', fps=5)
156
  for im in all_frames:
157
  writer.append_data(np.array(im))
158
  writer.close()
 
108
  def inference(text):
109
  all_frames = []
110
  target = clip_model.embed_text(text)
111
+ steps = 300
112
  seed = 2
113
  tf = Compose([
114
  Resize(224),
 
152
  all_frames.append(pil_image)
153
  #os.makedirs(f'samples/{timestring}', exist_ok=True)
154
  #pil_image.save(f'samples/{timestring}/{i:04}.jpg')
155
+ writer = imageio.get_writer('test.mp4', fps=15)
156
  for im in all_frames:
157
  writer.append_data(np.array(im))
158
  writer.close()