Update app.py

app.py

@@ -278,7 +278,9 @@ def norm_path(filepath):
     img = np.clip(img, 0, 1)
     fpath, fext = os.path.splitext(filepath)
     filepath = fpath +'.png'
-    imsave(filepath, img)
+    pil_image = Image.fromarray(img)
+    pil_image.save(filepath)
+    #imsave(filepath, pil_image)
     return filepath 
     
 def update_image(filepath):