update

app.py

@@ -312,14 +312,14 @@ def main(args):
     ]
     # outputs
     outputs = gr.outputs.Image(type="numpy", label="Vectorized Image")
-    outputs02 = gr.outputs.JSON(label="Iterations")
+    outputs02 = gr.outputs.File(label="Generated SVG output")
 
     # title
     title = "LIVE: Towards Layer-wise Image Vectorization"
     # description
     description = "<div align='center'>(CVPR 2022 Oral Presentation)</div>" \
                   "<div align='center'>Without GPUs, LIVE will cost longer time.</div>" \
-                  "<div align='center'>For efficiency, we rescale all input images to 160x160 in Huggingface Space.</div> "
+                  "<div align='center'>For efficiency, we rescale input to 160x160 (smaller size and fewer iterations will decrease the reconstructions).</div> "
 
     # examples
     examples = [

main.py

@@ -640,11 +640,13 @@ def main_func(target, experiment, num_iter, cfg_arg):
                 imshow = img.detach().cpu()
             pydiffvg.imwrite(imshow, filename, gamma=gamma)
 
+        svg_app_file_name = ""
         if cfg.save.output:
             filename = os.path.join(
                 cfg.experiment_dir, "output-svg", "{}.svg".format(pathn_record_str))
             check_and_create_dir(filename)
             pydiffvg.save_svg(filename, w, h, shapes_record, shape_groups_record)
+            svg_app_file_name = filename
 
         loss_matrix.append(loss_list)
 
@@ -693,7 +695,7 @@ def main_func(target, experiment, num_iter, cfg_arg):
             # shutil.rmtree(os.path.join(cfg.experiment_dir, "video-png"))
 
     print("The last loss is: {}".format(loss.item()))
-    return img.detach().cpu().numpy(), t
+    return img.detach().cpu().numpy(), svg_app_file_name
 
 
 if __name__ == "__main__":