Spaces:
Sleeping
Sleeping
gemini-2.0-flash added
Browse files
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 |
|