ZeyuXie commited on
Commit
9fd4c53
1 Parent(s): 629d1bf

Update llm_preprocess.py

Browse files
Files changed (1) hide show
  1. llm_preprocess.py +1 -0
llm_preprocess.py CHANGED
@@ -63,6 +63,7 @@ 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
 
63
  preffix_prompt = get_prompt()
64
  import google.generativeai as genai
65
  GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
66
+ print(GEMINI_API_KEY)
67
  genai.configure(api_key=GEMINI_API_KEY)
68
 
69
  # Set up the model