Fucius commited on
Commit
4104006
1 Parent(s): 4fc7cec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -367,8 +367,7 @@ def main(device, segment_type):
367
  image = Image.fromarray((image * 255.0).clip(0, 255).astype(np.uint8))
368
  return image
369
 
370
- # @spaces.GPU(duration=200)
371
- @spaces.GPU
372
  def generate_image(prompt1, negative_prompt, man, woman, resolution, local_prompt1, local_prompt2, seed, condition, condition_img1, style):
373
  # try:
374
  path1 = lorapath_man[man]
 
367
  image = Image.fromarray((image * 255.0).clip(0, 255).astype(np.uint8))
368
  return image
369
 
370
+ @spaces.GPU(duration=200)
 
371
  def generate_image(prompt1, negative_prompt, man, woman, resolution, local_prompt1, local_prompt2, seed, condition, condition_img1, style):
372
  # try:
373
  path1 = lorapath_man[man]