Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ os.environ["CEREBRAS_API_KEY"] = api_key
|
|
27 |
llm = Cerebras(model="llama-3.3-70b", api_key=os.environ["CEREBRAS_API_KEY"]) # Change model to Llama3.1-70b from Cerebras
|
28 |
Settings.llm = llm # Ensure Cerebras is the LLM being used
|
29 |
|
30 |
-
# Initialize
|
31 |
mixedbread_api_key = os.getenv("MXBAI_API_KEY")
|
32 |
embed_model = MixedbreadAIEmbedding(api_key=mixedbread_api_key, model_name="mixedbread-ai/mxbai-embed-large-v1")
|
33 |
|
|
|
27 |
llm = Cerebras(model="llama-3.3-70b", api_key=os.environ["CEREBRAS_API_KEY"]) # Change model to Llama3.1-70b from Cerebras
|
28 |
Settings.llm = llm # Ensure Cerebras is the LLM being used
|
29 |
|
30 |
+
# Initialize Mixedbread Embedding model
|
31 |
mixedbread_api_key = os.getenv("MXBAI_API_KEY")
|
32 |
embed_model = MixedbreadAIEmbedding(api_key=mixedbread_api_key, model_name="mixedbread-ai/mxbai-embed-large-v1")
|
33 |
|