Sanket commited on
Commit
c995500
·
1 Parent(s): 4e0eaa4
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -128,7 +128,7 @@ def predict(input_img, ver):
128
  Green = source[G].point(lambda i: i < 200 & i / constant)
129
  Blue = source[B].point(lambda i: i < 200 & i / constant)
130
 
131
- im_output = Image.merge(drawing.mode, (Red, Green, Blue))
132
  return im_output
133
 
134
 
 
128
  Green = source[G].point(lambda i: i < 200 & i / constant)
129
  Blue = source[B].point(lambda i: i < 200 & i / constant)
130
 
131
+ im_output = Image.merge(im_output.mode, (Red, Green, Blue))
132
  return im_output
133
 
134