Mattral commited on
Commit
ff2d69b
·
verified ·
1 Parent(s): 8626678

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -5,6 +5,8 @@ from langchain_community.document_loaders import PDFPlumberLoader
5
  from langchain.text_splitter import RecursiveCharacterTextSplitter
6
  from langchain_core.vectorstores import InMemoryVectorStore
7
  from langchain_core.prompts import ChatPromptTemplate
 
 
8
 
9
  # Set up Hugging Face model and token
10
  model = "mistralai/Mixtral-8x7B-Instruct-v0.1" # You can change to a model of your choice from Hugging Face
@@ -23,8 +25,11 @@ Answer:
23
  pdfs_directory = '../pdfs'
24
  os.makedirs(pdfs_directory, exist_ok=True)
25
 
 
 
 
26
  # Initialize the vector store for document indexing
27
- vector_store = InMemoryVectorStore()
28
 
29
  # Function to upload PDF file
30
  def upload_pdf(file):
 
5
  from langchain.text_splitter import RecursiveCharacterTextSplitter
6
  from langchain_core.vectorstores import InMemoryVectorStore
7
  from langchain_core.prompts import ChatPromptTemplate
8
+ from langchain.embeddings import OpenAIEmbeddings
9
+
10
 
11
  # Set up Hugging Face model and token
12
  model = "mistralai/Mixtral-8x7B-Instruct-v0.1" # You can change to a model of your choice from Hugging Face
 
25
  pdfs_directory = '../pdfs'
26
  os.makedirs(pdfs_directory, exist_ok=True)
27
 
28
+ # Initialize the embedding model
29
+ embedding = OpenAIEmbeddings()
30
+
31
  # Initialize the vector store for document indexing
32
+ vector_store = InMemoryVectorStore(embedding=embedding)
33
 
34
  # Function to upload PDF file
35
  def upload_pdf(file):