Hamam commited on
Commit
42cd559
·
verified ·
1 Parent(s): c73119c

update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -85,7 +85,9 @@ class ImageColorizationPipeline(object):
85
 
86
  return output_img
87
 
88
-
 
 
89
 
90
  def colorize_image(image):
91
  """Colorizes a grayscale image using the DDColor model."""
 
85
 
86
  return output_img
87
 
88
+ colorizer = ImageColorizationPipeline(model_path=model_path,
89
+ input_size=input_size,
90
+ model_size=model_size)
91
 
92
  def colorize_image(image):
93
  """Colorizes a grayscale image using the DDColor model."""