Spaces:

rein0421
/

AIdentify

Running

syurein commited on Jul 5

Commit

c2e61ba

1 Parent(s): 890ea7e

性能向上

Files changed (1) hide show

LLM_package.py CHANGED Viewed

@@ -21,7 +21,7 @@ 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-peo",
             contents=[my_file, prompt],
         )
         return response.text

         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",
             contents=[my_file, prompt],
         )
         return response.text