bug fix

app.py

@@ -66,7 +66,7 @@ pydiffvg.set_print_timing(False)
 gamma = 1.0
 
 
-def set_config(semantic_concept, word, font_name, num_steps):
+def set_config(semantic_concept, word, prompt, font_name, num_steps):
     
     cfg_d = edict()
     cfg_d.config = "code/config/base.yaml"
@@ -95,7 +95,7 @@ def set_config(semantic_concept, word, font_name, num_steps):
     
     if ' ' in cfg.word:
         raise gr.Error(f'should be only one word')
-    cfg.caption = f"a {cfg.semantic_concept}. {cfg.prompt_suffix}"
+    cfg.caption = prompt
     cfg.log_dir = f"output/{cfg.experiment}_{cfg.word}"
     if cfg.optimized_letter in cfg.word:
         cfg.optimized_letter = cfg.optimized_letter
@@ -142,12 +142,12 @@ def init_shapes(svg_path, trainable: Mapping[str, bool]):
     return shapes_init, shape_groups_init, parameters
 
 
-def run_main_ex(semantic_concept, word, font_name, num_steps):
-    return list(next(run_main_app(semantic_concept, word, font_name, num_steps, 1)))
+def run_main_ex(semantic_concept, word, prompt, font_name, num_steps):
+    return list(next(run_main_app(semantic_concept, word, prompt, font_name, num_steps, 1)))
                         
-def run_main_app(semantic_concept, word, font_name, num_steps, example=0):
+def run_main_app(semantic_concept, word, prompt, font_name, num_steps, example=0):
     
-    cfg = set_config(semantic_concept, word, font_name, num_steps)
+    cfg = set_config(semantic_concept, word, prompt, font_name, num_steps)
 
     pydiffvg.set_use_gpu(torch.cuda.is_available())
 
@@ -362,6 +362,7 @@ with gr.Blocks() as demo:
     inputs = [
         semantic_concept,
         word, 
+        prompt,
         font_name,
         num_steps
     ]