Ilyas KHIAT commited on
Commit
68af04f
·
1 Parent(s): 11aff05

upgrade to larger models

Browse files
Files changed (1) hide show
  1. RAG_PDF_WEB.py +3 -3
RAG_PDF_WEB.py CHANGED
@@ -45,17 +45,17 @@ def get_doc_chunks(docs):
45
  return docs
46
 
47
  def get_vectorstore_from_docs(doc_chunks):
48
- embedding = OpenAIEmbeddings(model="text-embedding-3-small")
49
  vectorstore = FAISS.from_documents(documents=doc_chunks, embedding=embedding)
50
  return vectorstore
51
 
52
  def get_vectorstore(text_chunks):
53
- embedding = OpenAIEmbeddings(model="text-embedding-3-small")
54
  vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embedding)
55
  return vectorstore
56
 
57
  def get_conversation_chain(vectorstore):
58
- llm = ChatOpenAI(model="gpt-3.5-turbo",temperature=0.5, max_tokens=2048)
59
  retriever=vectorstore.as_retriever()
60
  prompt = hub.pull("rlm/rag-prompt")
61
 
 
45
  return docs
46
 
47
  def get_vectorstore_from_docs(doc_chunks):
48
+ embedding = OpenAIEmbeddings(model="text-embedding-3-large")
49
  vectorstore = FAISS.from_documents(documents=doc_chunks, embedding=embedding)
50
  return vectorstore
51
 
52
  def get_vectorstore(text_chunks):
53
+ embedding = OpenAIEmbeddings(model="text-embedding-3-large")
54
  vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embedding)
55
  return vectorstore
56
 
57
  def get_conversation_chain(vectorstore):
58
+ llm = ChatOpenAI(model="gpt-4",temperature=0.5, max_tokens=2048)
59
  retriever=vectorstore.as_retriever()
60
  prompt = hub.pull("rlm/rag-prompt")
61