DanielIglesias97 commited on
Commit
95ec4d8
·
1 Parent(s): b6a5477

We have changed the app code to add the names of the images to the

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -1,20 +1,31 @@
1
  import configparser
2
  import gradio as gr
3
  import numpy as np
 
4
  from search_engine_model import SearchEngineModel
5
 
6
  def get_image_embeddings(input_image_paths_list):
7
  search_engine_model = SearchEngineModel()
8
 
9
  model, preprocess = search_engine_model.load_clip_model()
 
10
  image_embeddings_list = []
11
  for current_input_image_path_aux in input_image_paths_list:
12
  current_image_embeddings = search_engine_model.encode_image(model, preprocess, current_input_image_path_aux)
13
  image_embeddings_list.append(current_image_embeddings.values[0])
14
 
15
  image_embeddings_np = np.array(image_embeddings_list)
 
 
 
16
 
17
- return image_embeddings_np
 
 
 
 
 
 
18
 
19
  def main():
20
  config_manager_obj = configparser.ConfigParser()
 
1
  import configparser
2
  import gradio as gr
3
  import numpy as np
4
+ import pandas as pd
5
  from search_engine_model import SearchEngineModel
6
 
7
  def get_image_embeddings(input_image_paths_list):
8
  search_engine_model = SearchEngineModel()
9
 
10
  model, preprocess = search_engine_model.load_clip_model()
11
+ images_paths_list = []
12
  image_embeddings_list = []
13
  for current_input_image_path_aux in input_image_paths_list:
14
  current_image_embeddings = search_engine_model.encode_image(model, preprocess, current_input_image_path_aux)
15
  image_embeddings_list.append(current_image_embeddings.values[0])
16
 
17
  image_embeddings_np = np.array(image_embeddings_list)
18
+ image_embeddings_df = pd.DataFrame(image_embeddings_np)
19
+ image_embeddings_df.insert(0, "image_name", input_image_paths_list)
20
+ image_embeddings_np = image_embeddings_df.values
21
 
22
+ output_df = gr.DataFrame(
23
+ type="numpy",
24
+ headers=['image_name'] + [f'feature_{it}' for it in range(0, len(image_embeddings_df.columns)-1)],
25
+ value=image_embeddings_np
26
+ )
27
+
28
+ return output_df
29
 
30
  def main():
31
  config_manager_obj = configparser.ConfigParser()