Diangle commited on
Commit
4e676cc
1 Parent(s): 9c87410

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -44,21 +44,21 @@ HTML="""
44
  """
45
 
46
  DESCRIPTION="""Welcome to our video retrieval demo powered by [Searchium-ai/clip4clip-webvid150k](https://huggingface.co/Searchium-ai/clip4clip-webvid150k)! <br>
47
- Using free text search - you will find the top 5 most relevant clips among a dataset of 5.5 million video clips. <br>
48
  Discover, explore, and enjoy the world of video search at your fingertips.
49
  """
50
  ENDING = """For search acceleration capabilities, please refer to [Searchium.ai](https://www.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:
@@ -145,7 +145,7 @@ with gr.Blocks(theme=gr.themes.Default(spacing_size=gr.themes.sizes.spacing_lg,
145
  inp = gr.Textbox(placeholder="Write a sentence.")
146
  btn = gr.Button(value="Search")
147
  ex = [["natural wonders of the world"],["baking chocolate cake"],
148
- ["birds fly in the sky"], ["yoga routines for morning energy"]]
149
  gr.Examples(examples=ex,
150
  inputs=[inp]
151
  )
 
44
  """
45
 
46
  DESCRIPTION="""Welcome to our video retrieval demo powered by [Searchium-ai/clip4clip-webvid150k](https://huggingface.co/Searchium-ai/clip4clip-webvid150k)! <br>
47
+ Using free text search - you will find the top 5 most relevant clips among a dataset of 1.5 million video clips. <br>
48
  Discover, explore, and enjoy the world of video search at your fingertips.
49
  """
50
  ENDING = """For search acceleration capabilities, please refer to [Searchium.ai](https://www.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:
 
145
  inp = gr.Textbox(placeholder="Write a sentence.")
146
  btn = gr.Button(value="Search")
147
  ex = [["natural wonders of the world"],["baking chocolate cake"],
148
+ ["birds fly in the sky"]]
149
  gr.Examples(examples=ex,
150
  inputs=[inp]
151
  )