muhammadsalmanalfaridzi commited on
Commit
ed72e23
·
verified ·
1 Parent(s): 2ab080a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -12
app.py CHANGED
@@ -28,18 +28,8 @@ llm = Cerebras(model="llama-3.3-70b", api_key=os.environ["CEREBRAS_API_KEY"]) #
28
  Settings.llm = llm # Ensure Cerebras is the LLM being used
29
 
30
  # Initialize Voyage Embedding model
31
- voyage_api_key = os.getenv("VOYAGE_API_KEY")
32
- if not voyage_api_key:
33
- raise ValueError("VOYAGE_API_KEY is not set.")
34
- else:
35
- print("Voyage API key loaded successfully.")
36
-
37
- model_name = "voyage-multimodal-3"
38
- embed_model = VoyageEmbedding(
39
- model_name=model_name,
40
- voyage_api_key=voyage_api_key,
41
- truncation=True
42
- )
43
 
44
  # Initialize Groq client for Whisper Large V3
45
  groq_api_key = os.getenv("GROQ_API_KEY")
 
28
  Settings.llm = llm # Ensure Cerebras is the LLM being used
29
 
30
  # Initialize Voyage 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
 
34
  # Initialize Groq client for Whisper Large V3
35
  groq_api_key = os.getenv("GROQ_API_KEY")