multimodalart HF staff commited on
Commit
ebc29a2
1 Parent(s): f56a518

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -115,7 +115,8 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
115
  output_type="pil",
116
  ).images[0]
117
  return final_image
118
-
 
119
  def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
120
  if selected_index is None:
121
  raise gr.Error("You must select a LoRA before proceeding.")
 
115
  output_type="pil",
116
  ).images[0]
117
  return final_image
118
+
119
+ @spaces.GPU(duration=70)
120
  def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
121
  if selected_index is None:
122
  raise gr.Error("You must select a LoRA before proceeding.")