Barak1 commited on
Commit
3a410f0
·
2 Parent(s): 16858f2 698ea30

Merge branch 'main' of https://huggingface.co/spaces/barakmeiri/RNRI

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
 
@@ -134,7 +134,7 @@ with gr.Blocks(css="style.css") as demo:
134
  return image
135
 
136
 
137
- # @spaces.GPU
138
  def edit(inversion_state, target_prompt):
139
  if inversion_state is None:
140
  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
 
 
134
  return image
135
 
136
 
137
+ @spaces.GPU
138
  def edit(inversion_state, target_prompt):
139
  if inversion_state is None:
140
  raise gr.Error("Set inputs before editing.")