arjunanand13 commited on
Commit
650998e
1 Parent(s): b8db93d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,8 +38,8 @@ class Chatbot:
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
  # 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")
 
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
  # 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")