anurag-chand commited on
Commit
ff77898
Β·
verified Β·
1 Parent(s): ac48187

Deploy Vivekananda scriptures semantic search Gradio app

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 L2 distance
195
  distance = res.get("_distance", 1.0)
196
- similarity = max(0.0, 1.0 - (distance / 2.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