Lesterchia174 commited on
Commit
7155429
·
verified ·
1 Parent(s): c1e2119

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -26,9 +26,19 @@ from langchain_core.documents import Document
26
  groq_api_key = os.getenv("GROQ_API_KEY")
27
 
28
  # Initialize Embeddings and ChromaDB
29
- embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
 
 
 
 
 
 
30
  vectorstore = Chroma(embedding_function=embedding_model)
31
 
 
 
 
 
32
  # Short-term memory for the LLM
33
  chat_memory = []
34
 
 
26
  groq_api_key = os.getenv("GROQ_API_KEY")
27
 
28
  # Initialize Embeddings and ChromaDB
29
+ try:
30
+ embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
31
+ except ImportError:
32
+ # Fallback if sentence-transformers is not available
33
+ print("sentence-transformers not available, trying alternative model...")
34
+ embedding_model = HuggingFaceEmbeddings(model_name="BAAI/bge-small-en-v1.5")
35
+
36
  vectorstore = Chroma(embedding_function=embedding_model)
37
 
38
+ # Initialize Embeddings and ChromaDB
39
+ #embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
40
+ #vectorstore = Chroma(embedding_function=embedding_model)
41
+
42
  # Short-term memory for the LLM
43
  chat_memory = []
44