barakmeiri commited on
Commit
698ea30
·
verified ·
1 Parent(s): e3bc468

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -89,7 +89,7 @@ with gr.Blocks(css="style.css") as demo:
89
  return image_editor, "Input has set!"
90
 
91
 
92
- # @spaces.GPU
93
  def set_pipe(inversion_state, input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
94
  num_inversion_steps=4, inversion_max_step=0.6, rnri_iterations=2, rnri_alpha=0.1, rnri_lr=0.2):
95
  if device == 'cuda':
@@ -129,7 +129,7 @@ with gr.Blocks(css="style.css") as demo:
129
  return image
130
 
131
 
132
- # @spaces.GPU
133
  def edit(inversion_state, target_prompt):
134
  if inversion_state is None:
135
  raise gr.Error("Set inputs before editing.")
 
89
  return image_editor, "Input has set!"
90
 
91
 
92
+ @spaces.GPU
93
  def set_pipe(inversion_state, input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
94
  num_inversion_steps=4, inversion_max_step=0.6, rnri_iterations=2, rnri_alpha=0.1, rnri_lr=0.2):
95
  if device == 'cuda':
 
129
  return image
130
 
131
 
132
+ @spaces.GPU
133
  def edit(inversion_state, target_prompt):
134
  if inversion_state is None:
135
  raise gr.Error("Set inputs before editing.")