1 files changed, 3 insertions, 3 deletions
diff --git a/test.py b/test.py
index 54b033c..4efddb8 100644
--- a/test.py
+++ b/test.py
@@ -63,15 +63,15 @@ if __name__ == '__main__':
             if last_im is not None:
                 frac = 4.
                 tmp_im = im.copy()
-                array_a = np.divide(im, frac)
-                array_b = np.divide(last_im, frac/(frac-1))
+                array_a = np.multiply(im, 1/frac)
+                array_b = np.multiply(last_im, (frac-1)/frac)
                 print(im.shape, array_a.shape, array_b.shape)
                 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, mode='RGB')
+            image_pil = Image.fromarray(im)
             image_pil.save(save_path)
             os.rename(save_path, final_path)
     webpage.save()