Spaces:

AIOmarRehan
/

Animal_Image_Classification_Using_InceptionV3

Sleeping

AIOmarRehan commited on 5 days ago

Commit

87d27f2

verified ·

1 Parent(s): 6cabf3e

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -13,7 +13,7 @@ model = tf.keras.models.load_model(MODEL_PATH)
 CLASS_NAMES = ["Cat", "Dog", "Snake"]
-def preprocess_image(img: Image.Image, target_size=(256, 256):
     img = img.convert("RGB")
     img = img.resize(target_size)
     img = np.array(img).astype("float32") / 255.0

 CLASS_NAMES = ["Cat", "Dog", "Snake"]
+def preprocess_image(img: Image.Image, target_size=(256, 256)):
     img = img.convert("RGB")
     img = img.resize(target_size)
     img = np.array(img).astype("float32") / 255.0