ryanrwatkins commited on
Commit
4590834
1 Parent(s): 87fb0c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -335,7 +335,7 @@ def instantiate_LLM(LLM_provider,api_key,temperature=0.5,top_p=0.95,model_name=N
335
  if LLM_provider == "OpenAI":
336
  llm = ChatOpenAI(
337
  api_key=api_key,
338
- model=model_name, # in ["gpt-3.5-turbo", "gpt-3.5-turbo-0125", "gpt-4-turbo-preview"]
339
  temperature=temperature,
340
  model_kwargs={
341
  "top_p": top_p
@@ -344,7 +344,7 @@ def instantiate_LLM(LLM_provider,api_key,temperature=0.5,top_p=0.95,model_name=N
344
  if LLM_provider == "Google":
345
  llm = ChatGoogleGenerativeAI(
346
  google_api_key=api_key,
347
- model=gemini-pro, # "gemini-pro"
348
  temperature=temperature,
349
  top_p=top_p,
350
  convert_system_message_to_human=True
 
335
  if LLM_provider == "OpenAI":
336
  llm = ChatOpenAI(
337
  api_key=api_key,
338
+ model="gpt-3.5-turbo", # in ["gpt-3.5-turbo", "gpt-3.5-turbo-0125", "gpt-4-turbo-preview"]
339
  temperature=temperature,
340
  model_kwargs={
341
  "top_p": top_p
 
344
  if LLM_provider == "Google":
345
  llm = ChatGoogleGenerativeAI(
346
  google_api_key=api_key,
347
+ model="gemini-pro", # "gemini-pro"
348
  temperature=temperature,
349
  top_p=top_p,
350
  convert_system_message_to_human=True