huchiahsi commited on
Commit
06f311b
·
verified ·
1 Parent(s): e90ea76

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -34,7 +34,7 @@ import google.generativeai as genai
34
 
35
  GOOGLE_API_KEY = os.environ.get('GOOGLE_API_KEY')
36
  genai.configure(api_key=GOOGLE_API_KEY)
37
- model = genai.GenerativeModel('gemini-1.5-flash')
38
 
39
  # API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
40
  def query(payload):
 
34
 
35
  GOOGLE_API_KEY = os.environ.get('GOOGLE_API_KEY')
36
  genai.configure(api_key=GOOGLE_API_KEY)
37
+ model = genai.GenerativeModel('gemini-2.0-flash')
38
 
39
  # API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
40
  def query(payload):