Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -65,6 +65,7 @@ If Cobra is helpful, please help to ⭐ the <a href='https://github.com/zhuang20
|
|
65 |
📧 **Contact**
|
66 |
<br>
|
67 |
If you have any questions, please feel free to reach me out at <b>[email protected]</b>.
|
|
|
68 |
📝 **Citation**
|
69 |
<br>
|
70 |
If our work is useful for your research, please consider citing:
|
@@ -433,7 +434,7 @@ def extract_sketch_line_image(query_image_, input_style):
|
|
433 |
|
434 |
return extracted_sketch_line.convert('RGB'), extracted_sketch_line.convert('RGB'), hint_mask, query_image_, extracted_sketch_line_ori.convert('RGB'), resolution
|
435 |
|
436 |
-
@spaces.GPU
|
437 |
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):
|
438 |
if extracted_line is None:
|
439 |
gr.Info("Please preprocess the image first")
|
|
|
65 |
📧 **Contact**
|
66 |
<br>
|
67 |
If you have any questions, please feel free to reach me out at <b>[email protected]</b>.
|
68 |
+
|
69 |
📝 **Citation**
|
70 |
<br>
|
71 |
If our work is useful for your research, please consider citing:
|
|
|
434 |
|
435 |
return extracted_sketch_line.convert('RGB'), extracted_sketch_line.convert('RGB'), hint_mask, query_image_, extracted_sketch_line_ori.convert('RGB'), resolution
|
436 |
|
437 |
+
@spaces.GPU(duration=180)
|
438 |
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):
|
439 |
if extracted_line is None:
|
440 |
gr.Info("Please preprocess the image first")
|