NHZ commited on
Commit
3fd2783
·
verified ·
1 Parent(s): 2ba3598

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -7,7 +7,6 @@ from sentence_transformers import SentenceTransformer
7
  from transformers import AutoTokenizer, AutoModel
8
  import torch
9
  from langchain.vectorstores import FAISS
10
- from langchain.embeddings import HuggingFaceEmbeddings
11
  from langchain.chains import RetrievalQA
12
  from langchain.prompts import PromptTemplate
13
  from langchain.llms.base import LLM
@@ -83,8 +82,8 @@ def create_vector_store(text):
83
  # Convert embeddings to a numpy array
84
  embeddings = np.array(embeddings, dtype=np.float32)
85
 
86
- # Create a FAISS vector store with valid embeddings
87
- vector_store = FAISS.from_texts(texts=sentences, embedding_function=lambda x: embeddings)
88
  return vector_store, sentences
89
 
90
  # Streamlit app
 
7
  from transformers import AutoTokenizer, AutoModel
8
  import torch
9
  from langchain.vectorstores import FAISS
 
10
  from langchain.chains import RetrievalQA
11
  from langchain.prompts import PromptTemplate
12
  from langchain.llms.base import LLM
 
82
  # Convert embeddings to a numpy array
83
  embeddings = np.array(embeddings, dtype=np.float32)
84
 
85
+ # Create a FAISS vector store with sentences and their embeddings
86
+ vector_store = FAISS.from_embeddings(embeddings=embeddings, texts=sentences)
87
  return vector_store, sentences
88
 
89
  # Streamlit app