JunhaoZhuang commited on
Commit
e83cf22
·
verified ·
1 Parent(s): 0cf3e95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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")