make api
Browse files- .gitignore +1 -1
- app.py +2 -2
.gitignore
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
.idea/
|
| 2 |
-
|
| 3 |
|
|
|
|
| 1 |
.idea/
|
| 2 |
+
__pycache__/
|
| 3 |
|
app.py
CHANGED
|
@@ -22,7 +22,7 @@ image_filter = ImageFilter(learn=learn)
|
|
| 22 |
colorizer = ModelImageColorizer(image_filter)
|
| 23 |
|
| 24 |
|
| 25 |
-
def
|
| 26 |
return colorizer.get_colored_image(image, render_factor=35)
|
| 27 |
|
| 28 |
|
|
@@ -46,7 +46,7 @@ This demo is running on a CPU, if you like this project please make us a donatio
|
|
| 46 |
"""
|
| 47 |
|
| 48 |
demo = gr.Interface(
|
| 49 |
-
|
| 50 |
gr.Image(type="pil", label="Image gray scale"),
|
| 51 |
], [
|
| 52 |
gr.Image(type="pil", label="Image color")
|
|
|
|
| 22 |
colorizer = ModelImageColorizer(image_filter)
|
| 23 |
|
| 24 |
|
| 25 |
+
def predict(image):
|
| 26 |
return colorizer.get_colored_image(image, render_factor=35)
|
| 27 |
|
| 28 |
|
|
|
|
| 46 |
"""
|
| 47 |
|
| 48 |
demo = gr.Interface(
|
| 49 |
+
predict, [
|
| 50 |
gr.Image(type="pil", label="Image gray scale"),
|
| 51 |
], [
|
| 52 |
gr.Image(type="pil", label="Image color")
|