Diangle commited on
Commit
757b931
·
verified ·
1 Parent(s): 737943e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -58,19 +58,18 @@ ENDING = """For search acceleration capabilities, please refer to [Searchium.ai]
58
 
59
 
60
 
61
- DATA_PATH = './cache'
62
-
63
  huggingface_hub.snapshot_download(repo_id="Searchium-ai/clip4clip-webvid150k", repo_type="model", cache_dir=DATA_PATH)
64
 
65
- ft_visual_features_file = DATA_PATH + '/new_data/video_half_dataset_visual_features.npy'
66
- ft_visual_features_file_bin = DATA_PATH + '/new_data/video_half_dataset_visual_features_binary_packed.npy'
67
 
68
  #load database features:
69
  ft_visual_features_database_bin = np.load(ft_visual_features_file_bin)
70
  ft_visual_features_database = np.load(ft_visual_features_file, mmap_mode='r')
71
 
72
 
73
- database_csv_path = os.path.join(DATA_PATH, 'video_half_dataset.csv')
74
  database_df = pd.read_csv(database_csv_path)
75
 
76
  class NearestNeighbors:
 
58
 
59
 
60
 
61
+ DATA_PATH = '/home/user'
 
62
  huggingface_hub.snapshot_download(repo_id="Searchium-ai/clip4clip-webvid150k", repo_type="model", cache_dir=DATA_PATH)
63
 
64
+ ft_visual_features_file = DATA_PATH + '/models--Searchium-ai--clip4clip-webvid150k/snapshots/d349063a151fe22e934d47d611a1ec97b8e6168e/new_data/video_half_dataset_visual_features.npy'
65
+ ft_visual_features_file_bin = DATA_PATH + '/models--Searchium-ai--clip4clip-webvid150k/snapshots/d349063a151fe22e934d47d611a1ec97b8e6168e/new_data/video_half_dataset_visual_features_binary_packed.npy'
66
 
67
  #load database features:
68
  ft_visual_features_database_bin = np.load(ft_visual_features_file_bin)
69
  ft_visual_features_database = np.load(ft_visual_features_file, mmap_mode='r')
70
 
71
 
72
+ database_csv_path = os.path.join(DATA_PATH, '/models--Searchium-ai--clip4clip-webvid150k/snapshots/d349063a151fe22e934d47d611a1ec97b8e6168e/new_data/video_half_dataset.csv')
73
  database_df = pd.read_csv(database_csv_path)
74
 
75
  class NearestNeighbors: