eogreen commited on
Commit
e309fdb
1 Parent(s): 97edb41

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,6 +17,7 @@ import os
17
  os.environ['OPENAI_API_KEY'] = "gl-U2FsdGVkX1+0bNWD6YsVLZUYsn0m1WfLxUzrP0xUFbtWFAfk9Z1Cz+mD8u1yqKtV"; # e.g. gl-U2FsdGVkX19oG1mRO+LGAiNeC7nAeU8M65G4I6bfcdI7+9GUEjFFbplKq48J83by
18
  os.environ["OPENAI_BASE_URL"] = "https://aibe.mygreatlearning.com/openai/v1" # e.g. "https://aibe.mygreatlearning.com/openai/v1";
19
 
 
20
  llm_client = OpenAI()
21
 
22
  # Define the embedding model and the vectorstore
@@ -142,7 +143,7 @@ def llm_query(user_input,company):
142
  return llm_response
143
 
144
  # Set-up the Gradio UI
145
- company = gr.Radio(label='Company:', choices=["aws", "google", "ibm", "meta", "microsoft"]) # Create a radio button for company selection
146
  textbox = gr.Textbox(label='Question:') # Create a textbox for user input
147
 
148
  # Create Gradio interface
 
17
  os.environ['OPENAI_API_KEY'] = "gl-U2FsdGVkX1+0bNWD6YsVLZUYsn0m1WfLxUzrP0xUFbtWFAfk9Z1Cz+mD8u1yqKtV"; # e.g. gl-U2FsdGVkX19oG1mRO+LGAiNeC7nAeU8M65G4I6bfcdI7+9GUEjFFbplKq48J83by
18
  os.environ["OPENAI_BASE_URL"] = "https://aibe.mygreatlearning.com/openai/v1" # e.g. "https://aibe.mygreatlearning.com/openai/v1";
19
 
20
+ model_name = 'gpt-4o-mini' # e.g. 'gpt-3.5-turbo'
21
  llm_client = OpenAI()
22
 
23
  # Define the embedding model and the vectorstore
 
143
  return llm_response
144
 
145
  # Set-up the Gradio UI
146
+ company = gr.Radio(label='Company:', choices=["aws", "google", "ibm", "meta", "microsoft"], value="aws") # Create a radio button for company selection
147
  textbox = gr.Textbox(label='Question:') # Create a textbox for user input
148
 
149
  # Create Gradio interface