Spaces:

ak0601
/

cat_dog_classifier

Runtime error

ak0601 commited on Dec 26, 2023

Commit

f4b14bc

1 Parent(s): bb7c351

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from PIL import Image as PIL_Image
 model = tf.keras.models.load_model('dogcat_model_bak.h5')
 def classify_image(input_image):
-    pil_img = PIL_Image.fromarray(input_image.astype('uint8'), 'RGB')
     # Preprocess the image
     img1 = pil_img.resize((64, 64))
     img2 = pil_img

 model = tf.keras.models.load_model('dogcat_model_bak.h5')
 def classify_image(input_image):
+    pil_img = PIL_Image.fromarray(input_image, 'RGB')
     # Preprocess the image
     img1 = pil_img.resize((64, 64))
     img2 = pil_img