fix bug at filter `match` function
Browse files- .gitignore +1 -0
- 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 += [
|
|
|
|
|
|
|
|
|
|
| 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
|
| 362 |
-
model_names=[os.path.basename(x
|
| 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 |
|