thenativefox commited on
Commit
285b311
·
1 Parent(s): dfb9a66

Fix the table name passed

Browse files
Files changed (1) hide show
  1. backend/semantic_search.py +4 -4
backend/semantic_search.py CHANGED
@@ -3,11 +3,11 @@ import os
3
  import gradio as gr
4
  from sentence_transformers import SentenceTransformer
5
 
6
- db = lancedb.connect(".lancedb")
7
 
8
- MODEL1_STRATEGY1 = os.path.join(".lancedb", "model1_fixed.lance")
9
- MODEL2_STRATEGY1 = os.path.join(".lancedb", "model2_fixed.lance")
10
- MODEL3_STRATEGY1 = os.path.join(".lancedb", "model3_fixed.lance")
11
 
12
  VECTOR_COLUMN = os.getenv("VECTOR_COLUMN", "vector")
13
  TEXT_COLUMN = os.getenv("TEXT_COLUMN", "text")
 
3
  import gradio as gr
4
  from sentence_transformers import SentenceTransformer
5
 
6
+ db = lancedb.connect("./.lancedb")
7
 
8
+ MODEL1_STRATEGY1 = os.path.join(".lancedb", "model1_fixed")
9
+ MODEL2_STRATEGY1 = os.path.join(".lancedb", "model2_fixed")
10
+ MODEL3_STRATEGY1 = os.path.join(".lancedb", "model3_fixed")
11
 
12
  VECTOR_COLUMN = os.getenv("VECTOR_COLUMN", "vector")
13
  TEXT_COLUMN = os.getenv("TEXT_COLUMN", "text")