changed visible params

app.py

@@ -152,7 +152,7 @@ def init_shapes(svg_path, trainable: Mapping[str, bool]):
 def run_main_ex(word, semantic_concept, num_steps):
     prompt = f"a {semantic_concept}. minimal flat 2d vector. lineal color. trending on artstation"
     font_name = "ArefRuqaa"
-    return list(next(run_main_app(semantic_concept, word, prompt, font_name, num_steps, 1)))
+    return list(next(run_main_app(semantic_concept, word, prompt, font_name, num_steps, 0)))
                         
 def run_main_app(semantic_concept, word, prompt, font_name, num_steps, example=0):
     
@@ -164,7 +164,7 @@ def run_main_app(semantic_concept, word, prompt, font_name, num_steps, example=0
     preprocess(cfg.font, cfg.word, cfg.optimized_letter, cfg.level_of_cc)
     filename_init = os.path.join("code/data/init/", f"{cfg.font}_{cfg.word}_scaled.svg").replace(" ", "_")
     if not example:
-        yield gr.update(value=filename_init,visible=True),gr.update(visible=False),gr.update(visible=False)
+        yield gr.update(value=filename_init,visible=True),gr.update(visible=True, label=f'iters: 0 / {cfg.num_iter}'),gr.update(visible=False)
 
     sds_loss = SDSLoss(cfg, device, model)
 
@@ -249,7 +249,7 @@ def run_main_app(semantic_concept, word, prompt, font_name, num_steps, example=0
     combine_word(cfg.word, cfg.optimized_letter, cfg.font, cfg.experiment_dir, device)
 
     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)
+    yield gr.update(value=filename_init,visible=True),gr.update(visible=True),gr.update(value=image,visible=True)
  
 
 def change_prompt(concept, prompt_suffix):
@@ -313,7 +313,7 @@ with gr.Blocks() as demo:
 
         with gr.Column():
             result0 = gr.Image(type="filepath", label="Initial Word").style(height=333)
-            result1 = gr.Image(type="filepath", label="Optimization Process").style(height=110)
+            result1 = gr.Image(type="filepath", label="Optimization Process").style(height=333)
             result2 = gr.Image(type="filepath", label="Final Result",visible=False).style(height=333)