push

1 files changed, 2 insertions, 2 deletions

diff --git a/test.py b/test.py
index ddea76a..9ffdf22 100644
--- a/test.py
+++ b/test.py
@@ -64,12 +64,12 @@ if __name__ == '__main__':
                 tmp_im = im.copy()
                 array_a = np.multiply(im, 1/frac)
                 array_b = np.multiply(last_im, (frac-1)/frac)
-                im = np.add(array_a, array_b)
+                im = np.add(array_a, array_b).clip(0, 255)
                 last_im = tmp_im
             else:
                 last_im = im.copy()
                 print(im.shape)
-            image_pil = Image.fromarray(im)
+            image_pil = Image.fromarray(im, mode='RGB')
             image_pil.save(save_path)
             os.rename(save_path, final_path)
     webpage.save()