Diangle commited on
Commit
5c912ec
·
1 Parent(s): 204e44a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -51,14 +51,14 @@ ENDING = """For search acceleration capabilities, please refer to [Searchium.ai]
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:
@@ -144,7 +144,8 @@ with gr.Blocks(theme=gr.themes.Default(spacing_size=gr.themes.sizes.spacing_lg,
144
  with gr.Column():
145
  inp = gr.Textbox(placeholder="Write a sentence.")
146
  btn = gr.Button(value="Search")
147
- ex = [["natural wonders of the world"],["birds fly in the sky"],["yoga routines for morning energy"],["baking chocolate cake"]]
 
148
  gr.Examples(examples=ex,
149
  inputs=[inp]
150
  )
 
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:
 
144
  with gr.Column():
145
  inp = gr.Textbox(placeholder="Write a sentence.")
146
  btn = gr.Button(value="Search")
147
+ ex = [["natural wonders of the world"],["birds fly in the sky"],
148
+ ["baking chocolate cake"]]
149
  gr.Examples(examples=ex,
150
  inputs=[inp]
151
  )