Spaces:
Runtime error
Runtime error
arjunanand13
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ class Chatbot:
|
|
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('
|
43 |
# cpu_index = faiss.read_index('faiss_index_new_model3.index')
|
44 |
# gpu_index = faiss.index_cpu_to_gpu(gpu_resource, 0, cpu_index)
|
45 |
print("Loaded embedding successfully")
|
|
|
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 |
# cpu_index = faiss.read_index('faiss_index_new_model3.index')
|
44 |
# gpu_index = faiss.index_cpu_to_gpu(gpu_resource, 0, cpu_index)
|
45 |
print("Loaded embedding successfully")
|