.

app.py

@@ -121,10 +121,9 @@ def predict(input_img, ver):
     # Making image darker
     source = drawing.split()
 
-    W = 0
     # constant by which each pixel is divided
 
-    White = source[W].point(lambda i: darken_pixel(i))
+    White = source.point(lambda i: darken_pixel(i))
 
     im_output = Image.merge(drawing.mode, (White))
     return im_output