syurein commited on
Commit
890ea7e
·
1 Parent(s): f8d2233

性能向上

Browse files
Files changed (1) hide show
  1. LLM_package.py +2 -2
LLM_package.py CHANGED
@@ -21,14 +21,14 @@ class GeminiInference:
21
  client = genai.Client(api_key=self.api_key_source)
22
  my_file = client.files.upload(file=file_path)
23
  response = client.models.generate_content(
24
- model="gemini-2.5-flash",
25
  contents=[my_file, prompt],
26
  )
27
  return response.text
28
  def get_response_text(self,prompt):
29
  client = genai.Client(api_key=self.api_key_source)
30
  response = client.models.generate_content(
31
- model="gemini-2.5-flash",
32
  contents=[prompt],
33
  )
34
  text = response.text
 
21
  client = genai.Client(api_key=self.api_key_source)
22
  my_file = client.files.upload(file=file_path)
23
  response = client.models.generate_content(
24
+ model="gemini-2.5-peo",
25
  contents=[my_file, prompt],
26
  )
27
  return response.text
28
  def get_response_text(self,prompt):
29
  client = genai.Client(api_key=self.api_key_source)
30
  response = client.models.generate_content(
31
+ model="gemini-2.5-pro",
32
  contents=[prompt],
33
  )
34
  text = response.text