Pavan2k4 commited on
Commit
30f1183
·
verified ·
1 Parent(s): 0337340

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,13 +12,13 @@ from huggingface_hub import login
12
 
13
  # Load environment variables
14
  load_dotenv()
15
- login(token=os.environ.get('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.environ.get('PINCE_CONE_LIGHT')
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.environ.get('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:
 
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: