Pavan2k4 commited on
Commit
e6ef252
·
verified ·
1 Parent(s): 72b0097

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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('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,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('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:
 
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: