fix: force conversion to RGB to solve runtime error

e2f6174b · Jonathan Passerat-Palmbach · 836ea750 · e2f6174b
Commit e2f6174b authored Apr 20, 2017 by Jonathan Passerat-Palmbach
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

nithumb.py nithumb.py +2 -0

No files found.
--- a/nithumb.py
+++ b/nithumb.py
@@ -16,4 +16,6 @@ image = load(parsed.inputfile).get_data()
 image = Image.fromarray(image[:, :, image.shape[2]//2] if image.ndim == 3
                        else image)
 image = image.resize([128, 128])
+if image.mode != 'RGB':
+  image = image.convert('RGB')
 image.save(parsed.thumbnail)