Update app.py

app.py

@@ -239,10 +239,13 @@ def run_main_app(semantic_concept, word, letter, font_name, num_steps, example=0
     image = os.path.join(cfg.experiment_dir,f"{cfg.font}_{cfg.word}_{cfg.optimized_letter}.svg")
     # yield gr.update(value=filename_init,visible=True),gr.update(visible=False),gr.update(value=image,visible=True)
 
-    with open(filename_init, "r") as f:
-        svg_content = f.read()
+    with open(filename_init, "r") as f0:
+        svg_content_init = f0.read()
+
+    with open(image, "r") as f1:
+        svg_content_final = f1.read()
         
-    yield gr.update(value=f"<div>{svg_content}</div>",visible=True),gr.update(visible=False),gr.update(value=image,visible=True)
+    yield gr.update(value=f"<div>{svg_content_init}</div>",label="Initial Word",height=333,visible=True),gr.update(visible=False),gr.update(value=f"<div>{svg_content_final}</div>",label="Final Result",height=333,visible=True)
  
 
 with gr.Blocks() as demo: