ZeyuXie commited on
Commit
594eb84
1 Parent(s): 5dbd795

Update llm_preprocess.py

Browse files
Files changed (1) hide show
  1. 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.environ["GEMINI_API_KEY"]
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