Keltezaa commited on
Commit
f7eb8d5
·
verified ·
1 Parent(s): dd4bd7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -341,7 +341,7 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
341
  return final_image
342
 
343
  @spaces.GPU(duration=75)
344
- def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
345
  if not selected_indices:
346
  raise gr.Error("You must select at least one LoRA before proceeding.")
347
 
@@ -400,7 +400,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
400
 
401
  # Generate image
402
  if image_input is not None:
403
- final_image = generate_image_to_image(prompt_mash, image_input, image_strength, steps, cfg_scale, width, height, seed)
404
  yield final_image, seed, gr.update(visible=False)
405
  else:
406
  image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
 
341
  return final_image
342
 
343
  @spaces.GPU(duration=75)
344
+ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
345
  if not selected_indices:
346
  raise gr.Error("You must select at least one LoRA before proceeding.")
347
 
 
400
 
401
  # Generate image
402
  if image_input is not None:
403
+ final_image = generate_image_to_image(prompt_mash, steps, cfg_scale, width, height, seed)
404
  yield final_image, seed, gr.update(visible=False)
405
  else:
406
  image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)