Spaces:
Running
Running
syurein
commited on
Commit
·
890ea7e
1
Parent(s):
f8d2233
性能向上
Browse files- 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-
|
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-
|
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
|