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

fix error in pickle

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