mulasagg commited on
Commit
8aec9fe
·
1 Parent(s): eb13a14
Files changed (1) hide show
  1. src/utils/get_features.py +1 -1
src/utils/get_features.py CHANGED
@@ -12,7 +12,7 @@ def get_img_api(hash, data_path):
12
  return None, None
13
 
14
  # Extract the image path
15
- img_path = 'src/data/images/' + row['SHA256'].values[0] + '.jpg'
16
 
17
  # Extract the API calls
18
  api_columns = df.columns[2:] # Skip the first two columns (SHA256 and Type)
 
12
  return None, None
13
 
14
  # Extract the image path
15
+ img_path = 'src/data/Images/' + row['SHA256'].values[0] + '.jpg'
16
 
17
  # Extract the API calls
18
  api_columns = df.columns[2:] # Skip the first two columns (SHA256 and Type)