cache fix

app.py

@@ -234,13 +234,14 @@ def run_main_app(semantic_concept, word, script, prompt_suffix, font_name, num_s
         img = render(w, h, 2, 2, step, None, *scene_args)
 
         # compose image with white background
-        img = img[:, :, 3:4] * img[:, :, :3] + torch.ones(img.shape[0], img.shape[1], 3, device=device) * (
-                    1 - img[:, :, 3:4])
+        img = img[:, :, 3:4] * img[:, :, :3] + torch.ones(img.shape[0], img.shape[1], 3, device=device) * (1 - img[:, :, 3:4])
         img = img[:, :, :3]
 
+        print(img.shape)
+
         if step % skip == 0:
             img_numpy = img.detach().cpu().numpy()*255
-            img_numpy = np.resize(img_numpy, (200, 200))
+            img_numpy = np.resize(img_numpy, (200, 200, 3))
             gif_frames += [img_numpy]
         
         filename = os.path.join(cfg.experiment_dir, "video-svg", f"iter{step:04d}.svg")
@@ -268,13 +269,9 @@ def run_main_app(semantic_concept, word, script, prompt_suffix, font_name, num_s
         scheduler.step()
 
             
-    filename = os.path.join(
-        cfg.experiment_dir, "output-svg", "output.svg")
+    filename = os.path.join(cfg.experiment_dir, "output-svg", "output.svg")
     check_and_create_dir(filename)
-    save_svg.save_svg(
-        filename, w, h, shapes, shape_groups)
-
-    # combine_word(cfg.word, cfg.optimized_letter, cfg.font, cfg.experiment_dir, device)
+    save_svg.save_svg(filename, w, h, shapes, shape_groups)
 
     filename = os.path.join(cfg.experiment_dir, "final.gif")
     # writer = imageio.get_writer(filename, fps=20)