Update app.py

app.py

@@ -12,11 +12,11 @@ def predict(image):
         # Convert to grayscale
         img = ImageOps.grayscale(image)
         
-        # Resize image to match the model's expected input size (280x280)
-        img = img.resize((280, 280))  # Adjusted the size to 280x280
+        # Resize image to match the model's expected input size (299x299)
+        img = img.resize((299, 299))  # Resize to 299x299
         
         # Convert to numpy array and normalize the pixel values
-        img_array = np.array(img).reshape(1, 280 * 280) / 255.0  # Flattened image to 78400
+        img_array = np.array(img).reshape(1, 299, 299, 1) / 255.0  # Reshape to (1, 299, 299, 1)
         
         # Make prediction
         prediction = model.predict(img_array)