Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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
|
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")
|