Maaz1 commited on
Commit
7fa09ff
·
verified ·
1 Parent(s): c2e726d

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +2 -2
config.py CHANGED
@@ -8,8 +8,8 @@ class Config:
8
 
9
  # Model settings
10
  VECTOR_DIMENSION = 1280
11
- INDEX_PATH = "models/trained_models/jewelry_index.idx"
12
- METADATA_PATH = "models/trained_models/jewelry_metadata.pkl"
13
 
14
  # Hardware settings
15
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
8
 
9
  # Model settings
10
  VECTOR_DIMENSION = 1280
11
+ INDEX_PATH = "models/jewelry_index.idx"
12
+ METADATA_PATH = "models/jewelry_metadata.pkl"
13
 
14
  # Hardware settings
15
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")