Update app.py

app.py

@@ -55,10 +55,13 @@ ENDING = """For search acceleration capabilities, please refer to [Searchium.ai]
             """
 
 
-DATA_PATH = './new_data'
+# DATA_PATH = './new_data'
+DATA_PATH = './data
 
-ft_visual_features_file = DATA_PATH + '/video_half_dataset_visual_features.npy'
-ft_visual_features_file_bin = DATA_PATH + '/video_half_dataset_visual_features_binary_packed.npy'
+# ft_visual_features_file = DATA_PATH + '/video_half_dataset_visual_features.npy'
+# ft_visual_features_file_bin = DATA_PATH + '/video_half_dataset_visual_features_binary_packed.npy'
+ft_visual_features_file = DATA_PATH + 'dataset_v1_visual_features.npy'
+ft_visual_features_file_bin = DATA_PATH + 'dataset_v1_visual_features_binary_packed.npy'
 
 #load database features:
 ft_visual_features_database_bin = np.load(ft_visual_features_file_bin)
@@ -66,7 +69,7 @@ ft_visual_features_database = np.memmap(ft_visual_features_file, dtype='float32'
                                         shape=(ft_visual_features_database_bin.shape[0], ft_visual_features_database_bin.shape[1]*8))
 
 
-database_csv_path = os.path.join(DATA_PATH, 'video_half_dataset.csv')
+database_csv_path = os.path.join(DATA_PATH, 'dataset_v1.csv')
 database_df = pd.read_csv(database_csv_path)
 
 class NearestNeighbors: