Spaces:
Runtime error
Runtime error
Commit
Β·
3055619
1
Parent(s):
2b16ad8
fixed the bugs
Browse files
app.py
CHANGED
|
@@ -125,7 +125,7 @@ def add_item(mode: str, uploaded_image, text_description: str):
|
|
| 125 |
# -------------------------
|
| 126 |
# App logic: search
|
| 127 |
# -------------------------
|
| 128 |
-
def search_items(query_image, query_text, limit: int = 5):
|
| 129 |
if query_image is not None:
|
| 130 |
qvec = embed_image_pil(query_image).tolist()
|
| 131 |
elif query_text and len(query_text.strip()) > 0:
|
|
@@ -143,11 +143,18 @@ def search_items(query_image, query_text, limit: int = 5):
|
|
| 143 |
|
| 144 |
results = []
|
| 145 |
for h in hits:
|
| 146 |
-
payload = h.payload or {}
|
| 147 |
score = getattr(h, "score", None)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 148 |
results.append(
|
| 149 |
-
f"id:{h.id} | score:{
|
| 150 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 151 |
return "\n\n".join(results)
|
| 152 |
|
| 153 |
# -------------------------
|
|
@@ -166,11 +173,13 @@ with gr.Blocks(title="Lost & Found β Simple Helper") as demo:
|
|
| 166 |
gr.Markdown("### π Search")
|
| 167 |
query_img = gr.Image(type="pil", label="Search by image (optional)")
|
| 168 |
query_text = gr.Textbox(lines=2, label="Search by text (optional)")
|
|
|
|
|
|
|
| 169 |
search_btn = gr.Button("π Search")
|
| 170 |
search_out = gr.Textbox(label="Search results", interactive=False)
|
| 171 |
|
| 172 |
add_btn.click(add_item, inputs=[mode, upload_img, text_desc], outputs=[add_out])
|
| 173 |
-
search_btn.click(search_items, inputs=[query_img, query_text], outputs=[search_out])
|
| 174 |
|
| 175 |
if __name__ == "__main__":
|
| 176 |
demo.launch(server_name="0.0.0.0", server_port=7860)
|
|
|
|
| 125 |
# -------------------------
|
| 126 |
# App logic: search
|
| 127 |
# -------------------------
|
| 128 |
+
def search_items(query_image, query_text, limit: int = 5, min_score: float = 0.90):
|
| 129 |
if query_image is not None:
|
| 130 |
qvec = embed_image_pil(query_image).tolist()
|
| 131 |
elif query_text and len(query_text.strip()) > 0:
|
|
|
|
| 143 |
|
| 144 |
results = []
|
| 145 |
for h in hits:
|
|
|
|
| 146 |
score = getattr(h, "score", None)
|
| 147 |
+
if score is None or score < min_score: # filter by threshold
|
| 148 |
+
continue
|
| 149 |
+
|
| 150 |
+
payload = h.payload or {}
|
| 151 |
results.append(
|
| 152 |
+
f"id:{h.id} | score:{score:.4f} | mode:{payload.get('mode','')} | tags:{payload.get('tags','')} | text:{payload.get('text','')}"
|
| 153 |
)
|
| 154 |
+
|
| 155 |
+
if not results:
|
| 156 |
+
return f"No results above similarity threshold {min_score:.2f}"
|
| 157 |
+
|
| 158 |
return "\n\n".join(results)
|
| 159 |
|
| 160 |
# -------------------------
|
|
|
|
| 173 |
gr.Markdown("### π Search")
|
| 174 |
query_img = gr.Image(type="pil", label="Search by image (optional)")
|
| 175 |
query_text = gr.Textbox(lines=2, label="Search by text (optional)")
|
| 176 |
+
limit_slider = gr.Slider(1, 10, value=5, step=1, label="Max results")
|
| 177 |
+
score_slider = gr.Slider(0.0, 1.0, value=0.90, step=0.01, label="Min similarity score")
|
| 178 |
search_btn = gr.Button("π Search")
|
| 179 |
search_out = gr.Textbox(label="Search results", interactive=False)
|
| 180 |
|
| 181 |
add_btn.click(add_item, inputs=[mode, upload_img, text_desc], outputs=[add_out])
|
| 182 |
+
search_btn.click(search_items, inputs=[query_img, query_text, limit_slider, score_slider], outputs=[search_out])
|
| 183 |
|
| 184 |
if __name__ == "__main__":
|
| 185 |
demo.launch(server_name="0.0.0.0", server_port=7860)
|