RohanSardar commited on
Commit
3d79f12
·
verified ·
1 Parent(s): 435219f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,7 +11,6 @@ from langchain_community.vectorstores import FAISS
11
  from langchain_community.document_loaders import PyPDFDirectoryLoader
12
 
13
  load_dotenv()
14
- os.environ["GROQ_API_KEY"] = os.getenv("GROQ_API_KEY")
15
  groq_api_key = os.getenv("GROQ_API_KEY")
16
 
17
  llm = ChatGroq(groq_api_key=groq_api_key, model="Llama3-8b-8192")
@@ -31,7 +30,7 @@ st.title("Simple RAG Application")
31
 
32
  def create_vector_embedding():
33
  if "vectors" not in st.session_state:
34
- st.session_state.embeddings = HuggingFaceBgeEmbeddings()
35
  st.session_state.loader = PyPDFDirectoryLoader("documents")
36
  st.session_state.docs = st.session_state.loader.load()
37
  st.session_state.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
 
11
  from langchain_community.document_loaders import PyPDFDirectoryLoader
12
 
13
  load_dotenv()
 
14
  groq_api_key = os.getenv("GROQ_API_KEY")
15
 
16
  llm = ChatGroq(groq_api_key=groq_api_key, model="Llama3-8b-8192")
 
30
 
31
  def create_vector_embedding():
32
  if "vectors" not in st.session_state:
33
+ st.session_state.embeddings = HuggingFaceBgeEmbeddings(model_name="all-MiniLM-L6-v2")
34
  st.session_state.loader = PyPDFDirectoryLoader("documents")
35
  st.session_state.docs = st.session_state.loader.load()
36
  st.session_state.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)