Spaces:

sgongora27
/

PAYADOR-experiments

Paused

sgongora27 commited on Dec 5, 2024

Commit

8a9c486

1 Parent(s): 911ab88

fix

Files changed (1) hide show

models.py CHANGED Viewed

@@ -28,7 +28,7 @@ class GeminiModel():
                 "threshold": "BLOCK_NONE",
             },
         ]
-        genai.configure(api_key=get_api_key(api_key_file))
         self.model = genai.GenerativeModel(model_name)
     def prompt_model(self,prompt: str) -> str:

                 "threshold": "BLOCK_NONE",
             },
         ]
+        genai.configure(api_key=os.getenv(api_key_file))
         self.model = genai.GenerativeModel(model_name)
     def prompt_model(self,prompt: str) -> str: