Spaces:
Sleeping
Sleeping
fix typo
Browse files
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/
|
| 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)
|