barakmeiri commited on
Commit
e74059b
·
verified ·
1 Parent(s): ebc25cc

Update app.py

Browse files

decorating edit() func.

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -42,6 +42,7 @@ def set_pipe(input_image, description_prompt, edit_guidance_scale, num_inference
42
  description_prompt, config, device,
43
  [rnri_iterations, rnri_alpha, rnri_lr]), "Input has set!"
44
 
 
45
  def edit(editor, target_prompt):
46
  if editor is None:
47
  raise gr.Error("Set inputs before editing.")
 
42
  description_prompt, config, device,
43
  [rnri_iterations, rnri_alpha, rnri_lr]), "Input has set!"
44
 
45
+ @spaces.GPU
46
  def edit(editor, target_prompt):
47
  if editor is None:
48
  raise gr.Error("Set inputs before editing.")