Update app.py

app.py

@@ -11,12 +11,16 @@ ImageFile.LOAD_TRUNCATED_IMAGES = True
 new_model = tf.keras.models.load_model('tree-not-tree.keras')
 
 def TreeOrNotTree(img):
-    img = img.resize((128, 128))
-    
-    # Convert the image to a numpy array and normalize it
-    img_array = np.array(img) / 255.0
+    # Convert PIL image to a numpy array (RGB format)
+    img_array = np.array(img)
+
+    # Resize the image to the expected input size (128x128) using cv2
+    img_array = cv2.resize(img_array, (128, 128))
+
+    # Normalize the image
+    img_array = img_array / 255.0
     
-    # Reshape to add batch dimension (1, 128, 128, 3) if needed
+    # Reshape to add batch dimension (1, 128, 128, 3)
     img_array = np.expand_dims(img_array, axis=0)
     
     # Make prediction