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