Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -58,19 +58,18 @@ ENDING = """For search acceleration capabilities, please refer to [Searchium.ai]
|
|
58 |
|
59 |
|
60 |
|
61 |
-
DATA_PATH = '
|
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:
|