Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,13 +12,13 @@ from huggingface_hub import login
|
|
12 |
|
13 |
# Load environment variables
|
14 |
load_dotenv()
|
15 |
-
login(token=os.
|
16 |
|
17 |
# Initialize Embedding Model
|
18 |
embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
19 |
|
20 |
# Pinecone Retriever
|
21 |
-
api_key = os.
|
22 |
if api_key is None:
|
23 |
raise ValueError("Pinecone API key missing.")
|
24 |
else:
|
@@ -36,7 +36,7 @@ llm_options = {
|
|
36 |
def load_llm(name, model_name):
|
37 |
"""Loads the selected LLM model only when needed."""
|
38 |
if name == "OpenAI":
|
39 |
-
openai_api_key = os.
|
40 |
return ChatOpenAI(model='gpt-4o-mini', openai_api_key=openai_api_key)
|
41 |
|
42 |
if "Phi" in name or "DeepSeek" in name:
|
|
|
12 |
|
13 |
# Load environment variables
|
14 |
load_dotenv()
|
15 |
+
login(token=os.getenv('HF_KEY'))
|
16 |
|
17 |
# Initialize Embedding Model
|
18 |
embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
19 |
|
20 |
# Pinecone Retriever
|
21 |
+
api_key = os.getenv('PINCE_CONE_LIGHT')
|
22 |
if api_key is None:
|
23 |
raise ValueError("Pinecone API key missing.")
|
24 |
else:
|
|
|
36 |
def load_llm(name, model_name):
|
37 |
"""Loads the selected LLM model only when needed."""
|
38 |
if name == "OpenAI":
|
39 |
+
openai_api_key = os.getenv('OPEN_AI_KEY')
|
40 |
return ChatOpenAI(model='gpt-4o-mini', openai_api_key=openai_api_key)
|
41 |
|
42 |
if "Phi" in name or "DeepSeek" in name:
|