Surbhi commited on
Commit
fef7bc5
·
1 Parent(s): 97322fb

fix error in pickle

Browse files
Files changed (1) hide show
  1. 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)