Sanket commited on
Commit
6944008
1 Parent(s): faafec4
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -121,14 +121,12 @@ def predict(input_img, ver):
121
  # Making image darker
122
  source = drawing.split()
123
 
124
- R, G, B = 0, 1, 2
125
  # constant by which each pixel is divided
126
 
127
- Red = source[R].point(lambda i: i / 2.0)
128
- Green = source[G].point(lambda i: i / 2.0)
129
- Blue = source[B].point(lambda i: i / 2.0)
130
 
131
- im_output = Image.merge(im_output.mode, (Red, Green, Blue))
132
  return im_output
133
 
134
 
 
121
  # Making image darker
122
  source = drawing.split()
123
 
124
+ W = 0
125
  # constant by which each pixel is divided
126
 
127
+ White = source[W].point(lambda i: darken_pixel(i))
 
 
128
 
129
+ im_output = Image.merge(im_output.mode, (White))
130
  return im_output
131
 
132