christian1984 commited on
Commit
7b4e8af
·
verified ·
1 Parent(s): 5e5a267

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
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
- vector_store = InMemoryVectorStore()
 
 
 
 
 
 
 
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):