Spaces:
Build error
Build error
Update app.py
Browse files
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)
|