fountai commited on
Commit
dc56d68
1 Parent(s): 63c64d0
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -99,6 +99,7 @@ pipe = FluxInpaintPipeline.from_pretrained(bfl_repo, torch_dtype=torch.bfloat16)
99
  MAX_SEED = np.iinfo(np.int32).max
100
  TRIGGER = "a photo of TOK"
101
 
 
102
  def execute(image, prompt):
103
  if not prompt :
104
  gr.Info("Please enter a text prompt.")
 
99
  MAX_SEED = np.iinfo(np.int32).max
100
  TRIGGER = "a photo of TOK"
101
 
102
+ @spaces.GPU()
103
  def execute(image, prompt):
104
  if not prompt :
105
  gr.Info("Please enter a text prompt.")