TheProjectsGuy commited on
Commit
1c2f636
1 Parent(s): b98feb8

Loading the extractor now (testing CPU load time)

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -71,9 +71,9 @@ assert os.path.isdir(cache_dir), "Cache directory not found"
71
  # %%
72
  # Model and transforms
73
  print("Loading DINO model")
74
- # extractor = DinoV2ExtractFeatures(dino_model, desc_layer, desc_facet,
75
- # device=device)
76
- extractor = None
77
  print("DINO model loaded")
78
  # VLAD path (directory)
79
  ext_s = f"{dino_model}/l{desc_layer}_{desc_facet}_c{num_c}"
 
71
  # %%
72
  # Model and transforms
73
  print("Loading DINO model")
74
+ # extractor = None
75
+ extractor = DinoV2ExtractFeatures(dino_model, desc_layer, desc_facet,
76
+ device=device)
77
  print("DINO model loaded")
78
  # VLAD path (directory)
79
  ext_s = f"{dino_model}/l{desc_layer}_{desc_facet}_c{num_c}"