Spaces:
Runtime error
Runtime error
Deploy Vivekananda scriptures semantic search Gradio app
Browse files
app.py
CHANGED
|
@@ -175,8 +175,8 @@ def semantic_search(query: str, limit: int = 5, min_similarity: float = 0.5) ->
|
|
| 175 |
# Encode query
|
| 176 |
query_vector = model.encode(query).tolist()
|
| 177 |
|
| 178 |
-
# Search LanceDB table
|
| 179 |
-
results = table.search(query_vector).limit(limit).to_list()
|
| 180 |
|
| 181 |
if not results:
|
| 182 |
return "<div style='text-align: center; color: #ff6b6b; padding: 20px;'>No results found. Try adjusting your query.</div>"
|
|
@@ -191,9 +191,9 @@ def semantic_search(query: str, limit: int = 5, min_similarity: float = 0.5) ->
|
|
| 191 |
except Exception:
|
| 192 |
pass
|
| 193 |
|
| 194 |
-
# Compute similarity from
|
| 195 |
distance = res.get("_distance", 1.0)
|
| 196 |
-
similarity = max(0.0, 1.0 -
|
| 197 |
|
| 198 |
if similarity < min_similarity:
|
| 199 |
continue
|
|
|
|
| 175 |
# Encode query
|
| 176 |
query_vector = model.encode(query).tolist()
|
| 177 |
|
| 178 |
+
# Search LanceDB table with Cosine Similarity
|
| 179 |
+
results = table.search(query_vector).metric("cosine").limit(limit).to_list()
|
| 180 |
|
| 181 |
if not results:
|
| 182 |
return "<div style='text-align: center; color: #ff6b6b; padding: 20px;'>No results found. Try adjusting your query.</div>"
|
|
|
|
| 191 |
except Exception:
|
| 192 |
pass
|
| 193 |
|
| 194 |
+
# Compute similarity from Cosine distance
|
| 195 |
distance = res.get("_distance", 1.0)
|
| 196 |
+
similarity = max(0.0, 1.0 - distance)
|
| 197 |
|
| 198 |
if similarity < min_similarity:
|
| 199 |
continue
|