bkhmsi commited on
Commit
6aafa80
1 Parent(s): 77fbb13

upscaled examples a bit

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -267,7 +267,7 @@ def run_main_app(semantic_concept, word, script, font_selected, prompt_suffix, n
267
  # training loop
268
  t_range = tqdm(range(num_iter))
269
  gif_frames = []
270
- skip = 5
271
  for step in t_range:
272
  optim.zero_grad()
273
 
@@ -289,7 +289,7 @@ def run_main_app(semantic_concept, word, script, font_selected, prompt_suffix, n
289
 
290
  if step % skip == 0:
291
  img_tensor = x.detach().cpu()
292
- img_tensor = torch.nn.functional.interpolate(img_tensor, size=(200, 200), mode='bilinear', align_corners=False)
293
  img_tensor = img_tensor.permute(0, 2, 3, 1).squeeze(0)
294
  gif_frames += [img_tensor.numpy()]
295
 
 
267
  # training loop
268
  t_range = tqdm(range(num_iter))
269
  gif_frames = []
270
+ skip = 10
271
  for step in t_range:
272
  optim.zero_grad()
273
 
 
289
 
290
  if step % skip == 0:
291
  img_tensor = x.detach().cpu()
292
+ img_tensor = torch.nn.functional.interpolate(img_tensor, size=(300, 300), mode='bilinear', align_corners=False)
293
  img_tensor = img_tensor.permute(0, 2, 3, 1).squeeze(0)
294
  gif_frames += [img_tensor.numpy()]
295