Sqxww commited on
Commit
20cc43d
·
1 Parent(s): 696e002

reuse edit prompt

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -23,6 +23,7 @@ extract_model_map = {
23
  def change_image_style(
24
  image_url,
25
  style_image_url,
 
26
  extract_model_tag,
27
  extract_prompt,
28
  prompt_prefix
@@ -92,11 +93,12 @@ def generate_image(
92
  continue
93
 
94
  for prompt_index in range(prompt_count):
 
95
  for image_index in range(image_count):
96
  completed_count += 1
97
  styleImageUrl = prefix + style_image[0]
98
  result_image, edit_prompt, error_msg = change_image_style(
99
- inputImageUrl, styleImageUrl, extract_model, extract_prompt, prompt_prefix
100
  )
101
 
102
  if result_image:
 
23
  def change_image_style(
24
  image_url,
25
  style_image_url,
26
+ edit_prompt,
27
  extract_model_tag,
28
  extract_prompt,
29
  prompt_prefix
 
93
  continue
94
 
95
  for prompt_index in range(prompt_count):
96
+ edit_prompt = ""
97
  for image_index in range(image_count):
98
  completed_count += 1
99
  styleImageUrl = prefix + style_image[0]
100
  result_image, edit_prompt, error_msg = change_image_style(
101
+ inputImageUrl, styleImageUrl, edit_prompt, extract_model, extract_prompt, prompt_prefix
102
  )
103
 
104
  if result_image: