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:
|