Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from PIL import Image
|
|
| 6 |
|
| 7 |
# --- Configuration ---
|
| 8 |
MODEL_PATH = "cats-vs-dogs-finetuned.keras"
|
| 9 |
-
IMAGE_SIZE = (
|
| 10 |
CLASS_LABELS = ['Cat', 'Dog']
|
| 11 |
|
| 12 |
# --- Load the Model ---
|
|
|
|
| 6 |
|
| 7 |
# --- Configuration ---
|
| 8 |
MODEL_PATH = "cats-vs-dogs-finetuned.keras"
|
| 9 |
+
IMAGE_SIZE = (180, 180) # Adjust this to match the input size your model expects!
|
| 10 |
CLASS_LABELS = ['Cat', 'Dog']
|
| 11 |
|
| 12 |
# --- Load the Model ---
|