Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,8 @@ from langchain_community.document_loaders import PDFPlumberLoader
|
|
5 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
6 |
from langchain_core.vectorstores import InMemoryVectorStore
|
7 |
from langchain_core.prompts import ChatPromptTemplate
|
|
|
|
|
8 |
|
9 |
HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
|
10 |
|
@@ -21,7 +23,14 @@ Context: {context}
|
|
21 |
Answer:
|
22 |
"""
|
23 |
|
24 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
|
26 |
|
27 |
def upload_pdf(file):
|
|
|
5 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
6 |
from langchain_core.vectorstores import InMemoryVectorStore
|
7 |
from langchain_core.prompts import ChatPromptTemplate
|
8 |
+
from langchain_community.vectorstores import InMemoryVectorStore
|
9 |
+
from langchain_huggingface import HuggingFaceEmbeddings # Import HF Embeddings
|
10 |
|
11 |
HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
|
12 |
|
|
|
23 |
Answer:
|
24 |
"""
|
25 |
|
26 |
+
|
27 |
+
|
28 |
+
# Initialize Hugging Face embeddings
|
29 |
+
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
30 |
+
|
31 |
+
# Initialize vector store with embeddings
|
32 |
+
vector_store = InMemoryVectorStore(embedding=embeddings)
|
33 |
+
|
34 |
|
35 |
|
36 |
def upload_pdf(file):
|