Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -51,14 +51,14 @@ ENDING = """For search acceleration capabilities, please refer to [Searchium.ai]
|
|
51 |
"""
|
52 |
|
53 |
|
54 |
-
DATA_PATH = './
|
55 |
|
56 |
-
ft_visual_features_file = DATA_PATH + '/
|
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, '
|
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"],
|
|
|
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 |
)
|