[CLEANUP][GEMINI]

1 year ago · fdc0163f7e
parent a01711fd23
commit fdc0163f7e
1 changed files with 1 additions and 3 deletions
--- a/playground/models/gemini.py
+++ b/playground/models/gemini.py
@ -7,9 +7,7 @@ load_dotenv()
 api_key = os.environ["GEMINI_API_KEY"]
 # Initialize the model
-model = Gemini(
+model = Gemini(gemini_api_key=api_key)
    gemini_api_key="AIzaSyCUMFvG3TwwuZIniSpvDeP2AW5USyJUgJM"
 )
 # Establish the prompt and image
 task = "What is your name"