Spaces:
Runtime error
Runtime error
Surbhi
commited on
Commit
·
fef7bc5
1
Parent(s):
97322fb
fix error in pickle
Browse files- utils/cnn.py +1 -1
utils/cnn.py
CHANGED
@@ -25,7 +25,7 @@ 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'), allow_pickle=True)
|
31 |
asins = list(asins)
|
|
|
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'), allow_pickle=True)
|
29 |
bottleneck_features_train = bottleneck_features_train.astype(np.float64)
|
30 |
asins = np.load(Config.read('app', 'cssasins'), allow_pickle=True)
|
31 |
asins = list(asins)
|