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
  ]