Lrosado commited on
Commit
327ec7c
·
verified ·
1 Parent(s): 7f371bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -18,15 +18,18 @@ from langchain_chroma import Chroma
18
  from huggingface_hub import CommitScheduler
19
  from pathlib import Path
20
 
 
 
 
21
  # Create Client
22
  client = OpenAI(
23
- api_key="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"
24
  )
25
 
26
  # Define the embedding model and the vectorstore
27
  #embedding_model = SentenceTransformerEmbeddings(model_name='thenlper/gte-small')
28
  #embedding_model = HuggingFaceEmbeddings(model_name='thenlper/gte-small')
29
- embedding_model = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_key="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")
30
  model_name = 'gpt-4o-mini'
31
  #Load the persisted vectorDB
32
  collection_name = '10k_embeddings'
 
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
+
24
  # Create Client
25
  client = OpenAI(
26
+ api_key="openai_api_key"
27
  )
28
 
29
  # Define the embedding model and the vectorstore
30
  #embedding_model = SentenceTransformerEmbeddings(model_name='thenlper/gte-small')
31
  #embedding_model = HuggingFaceEmbeddings(model_name='thenlper/gte-small')
32
+ embedding_model = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_key="openai_api_key")
33
  model_name = 'gpt-4o-mini'
34
  #Load the persisted vectorDB
35
  collection_name = '10k_embeddings'