Diangle commited on
Commit
ec84363
·
1 Parent(s): cd12225

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -51,14 +51,14 @@ ENDING = """For search acceleration capabilities, please refer to [Searchium.ai]
51
  """
52
 
53
 
54
- DATA_PATH = './data'
55
 
56
- ft_visual_features_file = DATA_PATH + '/dataset_v1_visual_features_database.npy'
57
 
58
  #load database features:
59
  ft_visual_features_database = np.load(ft_visual_features_file)
60
 
61
- database_csv_path = os.path.join(DATA_PATH, 'dataset_v1.csv')
62
  database_df = pd.read_csv(database_csv_path)
63
 
64
  class NearestNeighbors:
 
51
  """
52
 
53
 
54
+ DATA_PATH = './new_data'
55
 
56
+ ft_visual_features_file = DATA_PATH + '/video_dataset_visual_features_database.npy'
57
 
58
  #load database features:
59
  ft_visual_features_database = np.load(ft_visual_features_file)
60
 
61
+ database_csv_path = os.path.join(DATA_PATH, 'video_dataset.csv')
62
  database_df = pd.read_csv(database_csv_path)
63
 
64
  class NearestNeighbors: