JunhaoZhuang commited on
Commit
b7a3308
·
verified ·
1 Parent(s): c52d0eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -90,7 +90,7 @@ examples = [
90
  "./examples/shadow/example2/input.png",
91
  ["./examples/shadow/example2/reference_image_0.png"],
92
  "line + shadow", # style
93
- 1, # seed
94
  10, # step
95
  3, # top k
96
  ],
@@ -412,7 +412,7 @@ def extract_sketch_line_image(query_image_, input_style):
412
 
413
  return extracted_sketch_line.convert('RGB'), extracted_sketch_line.convert('RGB'), hint_mask, query_image_, extracted_sketch_line_ori.convert('RGB'), resolution
414
 
415
- @spaces.GPU(duration=60)
416
  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):
417
  if extracted_line is None:
418
  gr.Info("Please preprocess the image first")
 
90
  "./examples/shadow/example2/input.png",
91
  ["./examples/shadow/example2/reference_image_0.png"],
92
  "line + shadow", # style
93
+ 4, # seed
94
  10, # step
95
  3, # top k
96
  ],
 
412
 
413
  return extracted_sketch_line.convert('RGB'), extracted_sketch_line.convert('RGB'), hint_mask, query_image_, extracted_sketch_line_ori.convert('RGB'), resolution
414
 
415
+ @spaces.GPU
416
  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):
417
  if extracted_line is None:
418
  gr.Info("Please preprocess the image first")