Update app.py

app.py

@@ -78,7 +78,7 @@ def bot(
         response.resolve()
     else:
         image_prompt = preprocess_image(image_prompt)
-        model = genai.GenerativeModel('gemini-1.5-flash-latest')
+        model = genai.GenerativeModel(model_name)
         response = model.generate_content(
             contents=[text_prompt, image_prompt],
             stream=True,
@@ -94,7 +94,7 @@ def bot(
         response.resolve()
     else:
         #video_prompt = preprocess_image(video_prompt)
-        model = genai.GenerativeModel('gemini-1.5-flash-latest')
+        model = genai.GenerativeModel(model_name)
 
         video_file = genai.upload_file(path=video_prompt)
         while video_file.state.name == "PROCESSING":