Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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=
|
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'
|