VladB46 commited on
Commit
01df59e
·
verified ·
1 Parent(s): dfc7472

Update gemini_call.py

Browse files
Files changed (1) hide show
  1. gemini_call.py +1 -1
gemini_call.py CHANGED
@@ -66,7 +66,7 @@ final_prompt = f'''Task:
66
 
67
  api_key = os.getenv("GOOGLE_API_KEY")
68
  genai.configure(api_key=api_key)
69
- model = genai.GenerativeModel("gemini-2.0-flash-exp" , system_instruction=final_prompt)
70
 
71
  def extract_histogram_data(text):
72
  """Searches for reges to extract the info required for building the histogram"""
 
66
 
67
  api_key = os.getenv("GOOGLE_API_KEY")
68
  genai.configure(api_key=api_key)
69
+ model = genai.GenerativeModel("gemini-2.0-flash" , system_instruction=final_prompt)
70
 
71
  def extract_histogram_data(text):
72
  """Searches for reges to extract the info required for building the histogram"""