Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -369,9 +369,9 @@ def main_func(filepath):
|
|
369 |
model.setup_input(data)
|
370 |
model.optimize()
|
371 |
img=visualize(model, data, a)
|
372 |
-
img2=
|
373 |
-
img3=cv.resize(np.uint8(img), a, interpolation=cv.INTER_CUBIC)
|
374 |
-
img4=cv.resize(np.uint8(img), a, interpolation=cv.INTER_LINEAR)
|
375 |
return (size_text,img,img2,img3,img4)
|
376 |
|
377 |
title = "Image Colorization"
|
|
|
369 |
model.setup_input(data)
|
370 |
model.optimize()
|
371 |
img=visualize(model, data, a)
|
372 |
+
img2=np.resize(img,a)
|
373 |
+
img3=Image.fromarray(cv.resize(np.uint8(img), a, interpolation=cv.INTER_CUBIC))
|
374 |
+
img4=Image.fromarray(cv.resize(np.uint8(img), a, interpolation=cv.INTER_LINEAR))
|
375 |
return (size_text,img,img2,img3,img4)
|
376 |
|
377 |
title = "Image Colorization"
|