Spaces:
Running
Running
kingabzpro
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,8 @@ groq_api_key = os.getenv("Groq_API_Key")
|
|
14 |
llm = ChatGroq(model="llama-3.1-70b-versatile", api_key=groq_api_key)
|
15 |
|
16 |
# Initialize the embedding model
|
17 |
-
embed_model = HuggingFaceEmbeddings(model_name="mixedbread-ai/mxbai-embed-large-v1"
|
|
|
18 |
|
19 |
# Load the vector store from a local directory
|
20 |
vectorstore = Chroma(
|
|
|
14 |
llm = ChatGroq(model="llama-3.1-70b-versatile", api_key=groq_api_key)
|
15 |
|
16 |
# Initialize the embedding model
|
17 |
+
embed_model = HuggingFaceEmbeddings(model_name="mixedbread-ai/mxbai-embed-large-v1",
|
18 |
+
model_kwargs = {'device': 'cpu'})
|
19 |
|
20 |
# Load the vector store from a local directory
|
21 |
vectorstore = Chroma(
|