Hugging Face's logo

Spaces:
sampathlonka
/
veda_bot_2.0
Sleeping

App Files Community
samlonka commited on
Commit
0e51c38
·
1 Parent(s): f82c288

'credentials'

Browse files
Files changed (2) hide show
  1. Docs/ramana_docs_ids.pkl → ramana_docs_ids.pkl +0 -0
  2. vector_tool.py +1 -2
Docs/ramana_docs_ids.pkl → ramana_docs_ids.pkl RENAMED
File without changes
vector_tool.py CHANGED
@@ -18,7 +18,6 @@ load_dotenv()
18
  # Constants
19
  INDEX_NAME = "veda-index-v2"
20
  MODEL_NAME = "BAAI/bge-large-en-v1.5"
21
- DOCS_DIRECTORY = r"Docs/ramana_docs_ids.pkl"
22
 
23
 
24
 
@@ -48,7 +47,7 @@ def create_or_load_index():
48
  return pc.Index(INDEX_NAME)
49
 
50
  # Load documents
51
- docs = load_pickle(DOCS_DIRECTORY)
52
  # Initialize embedding model
53
  embedding = initialize_embedding_model(MODEL_NAME)
54
  # Create or load index
 
18
  # Constants
19
  INDEX_NAME = "veda-index-v2"
20
  MODEL_NAME = "BAAI/bge-large-en-v1.5"
 
21
 
22
 
23
 
 
47
  return pc.Index(INDEX_NAME)
48
 
49
  # Load documents
50
+ docs = load_pickle("ramana_docs_ids.pkl")
51
  # Initialize embedding model
52
  embedding = initialize_embedding_model(MODEL_NAME)
53
  # Create or load index