JunhaoZhuang commited on
Commit
9b67bab
·
verified ·
1 Parent(s): d31b15a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -413,7 +413,7 @@ def extract_sketch_line_image(query_image_, input_style):
413
 
414
  return extracted_sketch_line.convert('RGB'), extracted_sketch_line.convert('RGB'), hint_mask, query_image_, extracted_sketch_line_ori.convert('RGB'), resolution
415
 
416
- @spaces.GPU(duration=120)
417
  def colorize_image(input_style, extracted_line, reference_images, resolution, seed, num_inference_steps, top_k, hint_mask=None, hint_color=None, query_image_origin=None, extracted_image_ori=None):
418
  if extracted_line is None:
419
  gr.Info("Please preprocess the image first")
 
413
 
414
  return extracted_sketch_line.convert('RGB'), extracted_sketch_line.convert('RGB'), hint_mask, query_image_, extracted_sketch_line_ori.convert('RGB'), resolution
415
 
416
+ @spaces.GPU(duration=60)
417
  def colorize_image(input_style, extracted_line, reference_images, resolution, seed, num_inference_steps, top_k, hint_mask=None, hint_color=None, query_image_origin=None, extracted_image_ori=None):
418
  if extracted_line is None:
419
  gr.Info("Please preprocess the image first")