Spaces:
Runtime error
Runtime error
Commit
·
dc13855
1
Parent(s):
a8e3d05
app.py
Browse files
app.py
CHANGED
|
@@ -70,9 +70,11 @@ def dog_detector(img_path):
|
|
| 70 |
# Set PIL to be tolerant of image files that are truncated.
|
| 71 |
ImageFile.LOAD_TRUNCATED_IMAGES = True
|
| 72 |
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
|
|
|
|
|
|
| 76 |
|
| 77 |
# Transforms for the training, validation, and testing sets
|
| 78 |
#train_transforms = transforms.Compose([
|
|
|
|
| 70 |
# Set PIL to be tolerant of image files that are truncated.
|
| 71 |
ImageFile.LOAD_TRUNCATED_IMAGES = True
|
| 72 |
|
| 73 |
+
train_dir = 'dogImages/train'
|
| 74 |
+
test_dir = 'dogImages/test'
|
| 75 |
+
valid_dir = 'dogImages/valid'
|
| 76 |
+
|
| 77 |
+
train_folder = datasets.ImageFolder(train_dir)
|
| 78 |
|
| 79 |
# Transforms for the training, validation, and testing sets
|
| 80 |
#train_transforms = transforms.Compose([
|