Spaces:
Sleeping
Sleeping
Hamam
commited on
update app.py
Browse files
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."""
|