Spaces:

jkang
/

demo-artist-classifier

Runtime error

jaekookang commited on Jan 18, 2022

Commit

dbb7b85

1 Parent(s): 21c7de8

remove remove_alpha_channel

Files changed (2) hide show

.ipynb_checkpoints/gradio_artist_classifier-checkpoint.py CHANGED Viewed

@@ -43,7 +43,7 @@ def load_json_as_dict(json_file):
 def load_image_as_array(image_file):
     img = skimage.io.imread(image_file, as_gray=False, plugin='matplotlib')
-    if (img.shape[-1] > 3) & (remove_alpha_channel):  # if RGBA
         img = img[..., :-1]
     return img

 def load_image_as_array(image_file):
     img = skimage.io.imread(image_file, as_gray=False, plugin='matplotlib')
+    if (img.shape[-1] > 3):  # if RGBA
         img = img[..., :-1]
     return img

gradio_artist_classifier.py CHANGED Viewed

@@ -43,7 +43,7 @@ def load_json_as_dict(json_file):
 def load_image_as_array(image_file):
     img = skimage.io.imread(image_file, as_gray=False, plugin='matplotlib')
-    if (img.shape[-1] > 3) & (remove_alpha_channel):  # if RGBA
         img = img[..., :-1]
     return img

 def load_image_as_array(image_file):
     img = skimage.io.imread(image_file, as_gray=False, plugin='matplotlib')
+    if (img.shape[-1] > 3):  # if RGBA
         img = img[..., :-1]
     return img