Update app.py

app.py

@@ -84,6 +84,7 @@ def user(text_prompt: str, chatbot: CHAT_HISTORY):
 
 def bot(
  google_key: str,
+ model_name: str,
  files: Optional[List[str]],
  temperature: float,
  max_output_tokens: int,
@@ -114,14 +115,14 @@ def bot(
  if chatbot[-1][0] and isinstance(chatbot[-1][0], str) \
  else []
  image_prompt = [Image.open(file).convert('RGB') for file in files]
- model = genai.GenerativeModel('gemini-1.5-pro-exp-0801')
+ model = genai.GenerativeModel(model_name)
  response = model.generate_content(
  text_prompt + image_prompt,
  stream=True,
  generation_config=generation_config)
  else:
  messages = preprocess_chat_history(chatbot)
- model = genai.GenerativeModel('gemini-1.5-pro-exp-0801')
+ model = genai.GenerativeModel(model_name)
  response = model.generate_content(
  messages,
  stream=True,
@@ -136,6 +137,15 @@ def bot(
  time.sleep(0.01)
  yield chatbot
 
+model_selection = gr.Dropdown(
+ ["gemini-1.0-pro",
+ "gemini-1.5-flash",
+ "gemini-1.5-pro",
+ "gemini-1.5-pro-exp-0801"
+ ],
+ label="Select Gemini Model",
+ value="gemini-1.5-pro"
+)
 
 google_key_component = gr.Textbox(
  label="GOOGLE API KEY",
@@ -224,6 +234,7 @@ user_inputs = [
 
 bot_inputs = [
  google_key_component,
+ model_selection,
  upload_button_component,
  temperature_component,
  max_output_tokens_component,