NotASI commited on
Commit
72859da
·
1 Parent(s): e96ba45

Updated preview verion

Browse files
Files changed (1) hide show
  1. Tabs/Gemini_Chatbot_Preview.py +6 -2
Tabs/Gemini_Chatbot_Preview.py CHANGED
@@ -6,7 +6,7 @@ from dotenv import load_dotenv
6
  load_dotenv()
7
 
8
  GEMINI_API_KEY_PREVIEW = os.getenv("GEMINI_API_KEY_PREVIEW")
9
- model_name = "gemini-1.5-flash-exp-0827"
10
 
11
  genai.configure(api_key=GEMINI_API_KEY_PREVIEW)
12
 
@@ -96,7 +96,11 @@ gemini_chatbot_preview = gr.ChatInterface(
96
  clear_btn=clear_chat_button_preview,
97
  undo_btn=undo_chat_button_preview,
98
  additional_inputs=[
99
- gr.Dropdown(choices=["emini-1.5-pro", "gemini-1.5-pro-exp-0801", "gemini-1.5-pro-exp-0827", "gemini-1.5-flash", "gemini-1.5-flash-exp-0827", "gemini-1.5-flash-8b-exp-0827"], label="Models"),
 
 
 
 
100
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
101
  gr.Slider(minimum=1, maximum=8192, value=4096, step=1, label="Max new tokens"),
102
  gr.Slider(minimum=0.1, maximum=1.0, value=1, step=0.1, label="Temperature"),
 
6
  load_dotenv()
7
 
8
  GEMINI_API_KEY_PREVIEW = os.getenv("GEMINI_API_KEY_PREVIEW")
9
+ model_list = ["gemini-1.5-pro", "gemini-1.5-pro-exp-0801", "gemini-1.5-pro-exp-0827", "gemini-1.5-flash", "gemini-1.5-flash-exp-0827", "gemini-1.5-flash-8b-exp-0827"]
10
 
11
  genai.configure(api_key=GEMINI_API_KEY_PREVIEW)
12
 
 
96
  clear_btn=clear_chat_button_preview,
97
  undo_btn=undo_chat_button_preview,
98
  additional_inputs=[
99
+ gr.Dropdown(
100
+ choices=model_list,
101
+ value="gemini-1.5-flash-exp-0827",
102
+ label="Models"
103
+ ),
104
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
105
  gr.Slider(minimum=1, maximum=8192, value=4096, step=1, label="Max new tokens"),
106
  gr.Slider(minimum=0.1, maximum=1.0, value=1, step=0.1, label="Temperature"),