Update app.py

app.py

@@ -47,7 +47,7 @@ from moviepy.editor import VideoFileClip, concatenate_videoclips
 
 def bot(
     google_key: str,
-
+    model_name: str,
     image_prompt: Optional[Image.Image],
     video_prompt: List[Path],
     temperature: float,
@@ -83,7 +83,7 @@ def bot(
         else:
             video_path = str(video_prompt[0])
 
-        model = genai.GenerativeModel("gemini-1.5-pro-latest")
+        model = genai.GenerativeModel(model_name)
 
         video_file = genai.upload_file(path=video_path)
         while video_file.state.name == "PROCESSING":
@@ -102,14 +102,14 @@ def bot(
         response.resolve()        
     elif image_prompt is not None:
         image_prompt = preprocess_image(image_prompt)
-        model = genai.GenerativeModel("gemini-1.5-pro-latest")
+        model = genai.GenerativeModel(model_name)
         response = model.generate_content(
             contents=[text_prompt, image_prompt],
             stream=True,
             generation_config=generation_config)
         response.resolve()
     else:
-        model = genai.GenerativeModel("gemini-1.5-pro-latest")
+        model = genai.GenerativeModel(model_name)
         response = model.generate_content(
             text_prompt,
             stream=True,
@@ -213,6 +213,7 @@ user_inputs = [text_prompt_component,
 
 bot_inputs = [
     google_key_component,
+    model_selection,
     image_prompt_component,
     video_prompt_component,
     temperature_component,
@@ -229,6 +230,7 @@ with gr.Blocks() as demo:
     with gr.Column():
         google_key_component.render()
         with gr.Row():
+            model_selection.render()
             image_prompt_component.render()
             video_prompt_component.render()
             chatbot_component.render()