Lrosado commited on
Commit
defcfba
·
verified ·
1 Parent(s): 6cfd7e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -18,19 +18,21 @@ from langchain_chroma import Chroma
18
  from huggingface_hub import CommitScheduler
19
  from pathlib import Path
20
 
 
 
21
  # Retrieve the API key from the environment
22
  openai_api_key = os.getenv("OPENAI_API_KEY")
23
  print(f"API Key Loaded: {openai_api_key}") # Debugging Step
24
 
25
  # Create Client
26
  client = OpenAI(
27
- api_key="openai_api_key"
28
  )
29
 
30
  # Define the embedding model and the vectorstore
31
  #embedding_model = SentenceTransformerEmbeddings(model_name='thenlper/gte-small')
32
  #embedding_model = HuggingFaceEmbeddings(model_name='thenlper/gte-small')
33
- embedding_model = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_key="openai_api_key")
34
  model_name = 'gpt-4o-mini'
35
  #Load the persisted vectorDB
36
  collection_name = '10k_embeddings'
 
18
  from huggingface_hub import CommitScheduler
19
  from pathlib import Path
20
 
21
+
22
+
23
  # Retrieve the API key from the environment
24
  openai_api_key = os.getenv("OPENAI_API_KEY")
25
  print(f"API Key Loaded: {openai_api_key}") # Debugging Step
26
 
27
  # Create Client
28
  client = OpenAI(
29
+ api_key=openai_api_key
30
  )
31
 
32
  # Define the embedding model and the vectorstore
33
  #embedding_model = SentenceTransformerEmbeddings(model_name='thenlper/gte-small')
34
  #embedding_model = HuggingFaceEmbeddings(model_name='thenlper/gte-small')
35
+ embedding_model = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_key=openai_api_key)
36
  model_name = 'gpt-4o-mini'
37
  #Load the persisted vectorDB
38
  collection_name = '10k_embeddings'