fountai commited on
Commit
ef0e833
1 Parent(s): fae157c
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -98,7 +98,7 @@ pipe = FluxInpaintPipeline.from_pretrained(bfl_repo, torch_dtype=torch.bfloat16)
98
  MAX_SEED = np.iinfo(np.int32).max
99
  TRIGGER = "a photo of TOK"
100
 
101
- @spaces.GPU(duration=150)
102
  def execute(image, prompt):
103
  if not prompt :
104
  gr.Info("Please enter a text prompt.")
@@ -111,7 +111,7 @@ def execute(image, prompt):
111
  img = cv2.imread(image)
112
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
113
 
114
- imgs = [ random_positioning(img), random_positioning(img), random_positioning(img), random_positioning(img)]
115
 
116
  pipe.load_lora_weights("XLabs-AI/flux-RealismLora", weight_name='lora.safetensors')
117
  response = []
 
98
  MAX_SEED = np.iinfo(np.int32).max
99
  TRIGGER = "a photo of TOK"
100
 
101
+ @spaces.GPU(duration=200)
102
  def execute(image, prompt):
103
  if not prompt :
104
  gr.Info("Please enter a text prompt.")
 
111
  img = cv2.imread(image)
112
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
113
 
114
+ imgs = [ random_positioning(img), random_positioning(img)]
115
 
116
  pipe.load_lora_weights("XLabs-AI/flux-RealismLora", weight_name='lora.safetensors')
117
  response = []