Diangle commited on
Commit
b72a8fa
1 Parent(s): c62b1fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -53,10 +53,10 @@ ENDING = """For search acceleration capabilities, please refer to [Searchium.ai]
53
  """
54
 
55
 
56
- DATA_PATH = './data'
57
 
58
- ft_visual_features_file = DATA_PATH + '/dataset_v1_visual_features.npy'
59
- ft_visual_features_file_bin = DATA_PATH + '/dataset_v1_visual_features_binary_packed.npy'
60
 
61
  #load database features:
62
  ft_visual_features_database_bin = np.load(ft_visual_features_file_bin)
@@ -64,7 +64,7 @@ ft_visual_features_database = np.memmap(ft_visual_features_file, dtype='float32'
64
  shape=(ft_visual_features_database_bin.shape[0], ft_visual_features_database_bin.shape[1]*8))
65
 
66
 
67
- database_csv_path = os.path.join(DATA_PATH, 'dataset_v1.csv')
68
  database_df = pd.read_csv(database_csv_path)
69
 
70
  class NearestNeighbors:
 
53
  """
54
 
55
 
56
+ DATA_PATH = './new_data'
57
 
58
+ ft_visual_features_file = DATA_PATH + '/video_half_dataset_visual_features.npy'
59
+ ft_visual_features_file_bin = DATA_PATH + '/video_half_dataset_visual_features_binary_packed.npy'
60
 
61
  #load database features:
62
  ft_visual_features_database_bin = np.load(ft_visual_features_file_bin)
 
64
  shape=(ft_visual_features_database_bin.shape[0], ft_visual_features_database_bin.shape[1]*8))
65
 
66
 
67
+ database_csv_path = os.path.join(DATA_PATH, 'video_half_dataset.csv')
68
  database_df = pd.read_csv(database_csv_path)
69
 
70
  class NearestNeighbors: