Update app.py

app.py

@@ -370,8 +370,8 @@ def main_func(filepath):
         model.optimize()
         img=visualize(model, data, a)
         img2=np.resize(img,a)
-        img3=cv.resize(img, a, interpolation=cv.INTER_CUBIC)
-        img4=cv.resize(img, a, interpolation=cv.INTER_LINEAR)
+        img3=cv.resize(uint8(img), a, interpolation=cv.INTER_CUBIC)
+        img4=cv.resize(uint8(img), a, interpolation=cv.INTER_LINEAR)
     return (size_text,img,img2,img3,img4)
 
 title = "Image Colorization"