ramisn commited on
Commit
783143d
·
verified ·
1 Parent(s): 81149e0

gemini-2.0-flash added

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,7 +16,8 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
16
 
17
  # Function to load Google Gemini Model and provide queries as response
18
  def get_gemini_response(question, prompt):
19
- model = genai.GenerativeModel('gemini-pro')
 
20
  response = model.generate_content([prompt[0], question])
21
  return response.text.strip()
22
 
 
16
 
17
  # Function to load Google Gemini Model and provide queries as response
18
  def get_gemini_response(question, prompt):
19
+ # model = genai.GenerativeModel('gemini-pro')
20
+ model = genai.GenerativeModel("gemini-2.0-flash")
21
  response = model.generate_content([prompt[0], question])
22
  return response.text.strip()
23