🔧 Update generative model to 'gemini-1.5-flash' for improved performance in text and image generation functions.

Author: kkdai

main.py

@@ -1,4 +1,3 @@
-from linebot.models import FlexSendMessage
 from linebot.models import (
     MessageEvent, TextSendMessage
 )
@@ -99,7 +98,7 @@ def generate_gemini_text_complete(prompt):
     """
     Generate a text completion using the generative model.
     """
-    model = genai.GenerativeModel('gemini-pro')
+    model = genai.GenerativeModel('gemini-1.5-flash')
     response = model.generate_content(prompt)
     return response
 
@@ -109,7 +108,7 @@ def generate_result_from_image(img, prompt):
     Generate a image vision result using the generative model.
     """
 
-    model = genai.GenerativeModel('gemini-pro-vision')
+    model = genai.GenerativeModel('gemini-1.5-flash')
     response = model.generate_content([prompt, img], stream=True)
     response.resolve()
     return response