Update app.py
Browse files
app.py
CHANGED
|
@@ -626,7 +626,7 @@ transform = transforms.Compose([
|
|
| 626 |
transforms.Normalize(mean=mean, std=std)
|
| 627 |
])
|
| 628 |
|
| 629 |
-
test_dataset = datasets.ImageFolder(root='/
|
| 630 |
|
| 631 |
# Extracting features from Efficientnet model
|
| 632 |
def extract_features(test_dataset):
|
|
|
|
| 626 |
transforms.Normalize(mean=mean, std=std)
|
| 627 |
])
|
| 628 |
|
| 629 |
+
test_dataset = datasets.ImageFolder(root='/Datasets/Bananas', transform=transform)
|
| 630 |
|
| 631 |
# Extracting features from Efficientnet model
|
| 632 |
def extract_features(test_dataset):
|