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)