JiantaoLin commited on
Commit
8d61561
·
1 Parent(s): 155d1df
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -486,12 +486,14 @@ with gr.Blocks(css="""
486
 
487
  # Step 1: 点击 "Refine to detailed prompt" 按钮时
488
  def refine_to_detailed(prompt, seed):
 
489
  detailed_prompt_value = text_to_detailed(prompt, seed)
490
  generated_detailed_prompt = True
491
  return detailed_prompt_value
492
 
493
  # Step 2: 点击 "Generate Image" 按钮时
494
  def generate_image(prompt, seed):
 
495
  # 如果没有生成过详细描述,先执行 refine-to-detailed
496
  if not generated_detailed_prompt:
497
  detailed_prompt_value = text_to_detailed(prompt, seed)
@@ -504,6 +506,7 @@ with gr.Blocks(css="""
504
 
505
  # Step 3: 点击 "Generate Mesh" 按钮时
506
  def generate_mesh(image):
 
507
  # 如果没有生成图像,先执行生成图像
508
  if not generated_image:
509
  detailed_prompt_value = text_to_detailed(prompt, seed1)
 
486
 
487
  # Step 1: 点击 "Refine to detailed prompt" 按钮时
488
  def refine_to_detailed(prompt, seed):
489
+ nonlocal generated_detailed_prompt
490
  detailed_prompt_value = text_to_detailed(prompt, seed)
491
  generated_detailed_prompt = True
492
  return detailed_prompt_value
493
 
494
  # Step 2: 点击 "Generate Image" 按钮时
495
  def generate_image(prompt, seed):
496
+ nonlocal generated_image, generated_detailed_prompt
497
  # 如果没有生成过详细描述,先执行 refine-to-detailed
498
  if not generated_detailed_prompt:
499
  detailed_prompt_value = text_to_detailed(prompt, seed)
 
506
 
507
  # Step 3: 点击 "Generate Mesh" 按钮时
508
  def generate_mesh(image):
509
+ nonlocal generated_detailed_prompt, generated_image
510
  # 如果没有生成图像,先执行生成图像
511
  if not generated_image:
512
  detailed_prompt_value = text_to_detailed(prompt, seed1)