Spaces:
Sleeping
Sleeping
arjunanand13
commited on
Commit
•
9e36f19
1
Parent(s):
9961a38
Update app.py
Browse files
app.py
CHANGED
@@ -38,8 +38,8 @@ class Chatbot:
|
|
38 |
self.llm = HuggingFacePipeline(pipeline=self.generate_text)
|
39 |
|
40 |
try:
|
41 |
-
|
42 |
-
self.vectorstore = FAISS.load_local('faiss_index', HuggingFaceEmbeddings(model_name="flax-sentence-embeddings/all_datasets_v3_MiniLM-L12", model_kwargs={"device": "cuda"}))
|
43 |
print("Loaded embedding successfully")
|
44 |
except ImportError as e:
|
45 |
print("FAISS could not be imported. Make sure FAISS is installed correctly.")
|
|
|
38 |
self.llm = HuggingFacePipeline(pipeline=self.generate_text)
|
39 |
|
40 |
try:
|
41 |
+
self.vectorstore = FAISS.load_local('faiss_index', HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2", model_kwargs={"device": "cuda"}))
|
42 |
+
# self.vectorstore = FAISS.load_local('faiss_index', HuggingFaceEmbeddings(model_name="flax-sentence-embeddings/all_datasets_v3_MiniLM-L12", model_kwargs={"device": "cuda"}))
|
43 |
print("Loaded embedding successfully")
|
44 |
except ImportError as e:
|
45 |
print("FAISS could not be imported. Make sure FAISS is installed correctly.")
|