Spaces:
Running
on
Zero
Running
on
Zero
Update llm_preprocess.py
Browse files- llm_preprocess.py +1 -1
llm_preprocess.py
CHANGED
@@ -62,7 +62,7 @@ def postprocess(caption):
|
|
62 |
def preprocess_gemini(free_text_caption):
|
63 |
preffix_prompt = get_prompt()
|
64 |
import google.generativeai as genai
|
65 |
-
GEMINI_API_KEY = os.
|
66 |
genai.configure(api_key=GEMINI_API_KEY)
|
67 |
|
68 |
# Set up the model
|
|
|
62 |
def preprocess_gemini(free_text_caption):
|
63 |
preffix_prompt = get_prompt()
|
64 |
import google.generativeai as genai
|
65 |
+
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
66 |
genai.configure(api_key=GEMINI_API_KEY)
|
67 |
|
68 |
# Set up the model
|