性能向上

LLM_package.py

@@ -22,14 +22,14 @@ class GeminiInference:
         client = genai.Client(api_key=self.api_key_source)
         my_file = client.files.upload(file=file_path)
         response = client.models.generate_content(
-            model="gemini-2.5-pro",
+            model="gemini-2.5-flash",
             contents=[my_file, prompt],
         )
         return response.text
     def get_response_text(self,prompt):
         client = genai.Client(api_key=self.api_key_source)
         response = client.models.generate_content(
-            model="gemini-2.5-pro",
+            model="gemini-2.5-flash",
             contents=[prompt],
         )
         text = response.text