Spaces:
Runtime error
Runtime error
Surbhi
commited on
Commit
·
c09fd64
1
Parent(s):
fef7bc5
fix error in pickle
Browse files- utils/cnn.py +2 -2
utils/cnn.py
CHANGED
@@ -25,9 +25,9 @@ for col in required_columns:
|
|
25 |
raise ValueError(f"Missing required column: {col} in the dataset")
|
26 |
|
27 |
# Load the pre-trained CNN features and corresponding ASINs
|
28 |
-
bottleneck_features_train = np.load(Config.read('app', 'cnnmodel')
|
29 |
bottleneck_features_train = bottleneck_features_train.astype(np.float64)
|
30 |
-
asins = np.load(Config.read('app', 'cssasins')
|
31 |
asins = list(asins)
|
32 |
|
33 |
|
|
|
25 |
raise ValueError(f"Missing required column: {col} in the dataset")
|
26 |
|
27 |
# Load the pre-trained CNN features and corresponding ASINs
|
28 |
+
bottleneck_features_train = np.load(Config.read('app', 'cnnmodel'))
|
29 |
bottleneck_features_train = bottleneck_features_train.astype(np.float64)
|
30 |
+
asins = np.load(Config.read('app', 'cssasins'))
|
31 |
asins = list(asins)
|
32 |
|
33 |
|