Spaces:
Runtime error
Runtime error
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
|