whoami02 commited on
Commit
a23bd72
·
verified ·
1 Parent(s): 1b87df1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,13 +12,13 @@ from huggingface_hub import login
12
  from langchain.globals import set_verbose
13
  set_verbose(True)
14
 
15
- # load_dotenv(find_dotenv(r"E:\AW\LLMs\.env"))
16
 
17
  def load_model(model_id):
18
  if model_id == "gemini":
19
  return ChatGoogleGenerativeAI(
20
  model='gemini-pro',
21
- google_api_key=login(os.environ["GOOGLE_API_KEY"]),
22
  convert_system_message_to_human=True,
23
  temperature=0.05,
24
  verbose=True,
 
12
  from langchain.globals import set_verbose
13
  set_verbose(True)
14
 
15
+ # load_dotenv(find_dotenv(r".env"))
16
 
17
  def load_model(model_id):
18
  if model_id == "gemini":
19
  return ChatGoogleGenerativeAI(
20
  model='gemini-pro',
21
+ google_api_key=os.getenv["GOOGLE_API_KEY"],
22
  convert_system_message_to_human=True,
23
  temperature=0.05,
24
  verbose=True,