.

app.py

@@ -118,14 +118,10 @@ def predict(input_img, ver):
 
     drawing = transforms.ToPILImage()(drawing)
 
-    # Making image darker
-    source = drawing.split()
-
     # constant by which each pixel is divided
+    drawing = drawing.point(lambda i: darken_pixel(i))
 
-    White = source.point(lambda i: darken_pixel(i))
-
-    im_output = Image.merge(drawing.mode, (White))
+    im_output = drawing
     return im_output