TeeA commited on
Commit
9dbe966
·
1 Parent(s): af70222

fix bug at filter `match` function

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +8 -5
.gitignore CHANGED
@@ -35,6 +35,7 @@ pip-delete-this-directory.txt
35
  # Environment variables
36
  .env
37
  .gradio/
 
38
  # Local configuration files
39
  *.local
40
  # MacOS specific files
 
35
  # Environment variables
36
  .env
37
  .gradio/
38
+ gradio_cache/
39
  # Local configuration files
40
  *.local
41
  # MacOS specific files
app.py CHANGED
@@ -199,12 +199,15 @@ def reformat_and_return_top_k_results(
199
  result += model_names + [""] * (
200
  top_k - len(model_names)
201
  ) # Fill with empty string if less than top_k
202
- result += [f"Score: {score:.4f}" for score in similarity_scores] + [""] * (
 
 
 
203
  top_k - len(similarity_scores)
204
  ) # Fill with empty string if less than top_k
205
 
206
  logger.info(
207
- f"Found {len(model_names)} similar objects for the query: {model_names}"
208
  )
209
  return result
210
 
@@ -358,9 +361,9 @@ async def query_3D_object_by_hybrid_search_method(
358
 
359
  # Reformat and return top_k results
360
  return reformat_and_return_top_k_results(
361
- private_model_paths=[x[0] for x in matched_obj_paths[:top_k]],
362
- model_names=[os.path.basename(x[0]) for x in matched_obj_paths[:top_k]],
363
- similarity_scores=[None] * top_k,
364
  top_k=top_k,
365
  )
366
 
 
199
  result += model_names + [""] * (
200
  top_k - len(model_names)
201
  ) # Fill with empty string if less than top_k
202
+ result += [
203
+ "Score: " if not isinstance(score, float) else f"Score: {score:.4f}"
204
+ for score in similarity_scores
205
+ ] + [""] * (
206
  top_k - len(similarity_scores)
207
  ) # Fill with empty string if less than top_k
208
 
209
  logger.info(
210
+ f"Found {len(model_names)} similar objects for the query. They are: {model_names}"
211
  )
212
  return result
213
 
 
361
 
362
  # Reformat and return top_k results
363
  return reformat_and_return_top_k_results(
364
+ private_model_paths=[x for x in matched_obj_paths[:top_k]],
365
+ model_names=[os.path.basename(x) for x in matched_obj_paths[:top_k]],
366
+ similarity_scores=[None] * len(matched_obj_paths[:top_k]),
367
  top_k=top_k,
368
  )
369