Update app.py

app.py

@@ -16,7 +16,7 @@ def predict(image):
         img = img.resize((280, 280))  # Adjusted the size to 280x280
         
         # Convert to numpy array and normalize the pixel values
-        img_array = np.array(img).reshape(1, 280, 280, 1) / 255.0
+        img_array = np.array(img).reshape(1, 280 * 280) / 255.0  # Flattened image to 78400
         
         # Make prediction
         prediction = model.predict(img_array)