Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 87 |
-
vector_store = FAISS.
|
| 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
|