Update app.py

Aditya190803 · Sep 26, 2024 · a399b4b · a399b4b
1 parent 769eff1
commit a399b4b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -10,13 +10,13 @@
 # Define cached functions
 @st.cache_data()
 def get_gemini_response(input, pdf_content, prompt):
-    model = genai.GenerativeModel('gemini-pro-vision')
+    model = genai.GenerativeModel('gemini-1.5-flash')
     response = model.generate_content([input, pdf_content[0], prompt])
     return response.text
 
 @st.cache_data()
 def get_gemini_response_keywords(input, pdf_content, prompt):
-    model = genai.GenerativeModel('gemini-pro-vision')
+    model = genai.GenerativeModel('gemini-1.5-flash')
     response = model.generate_content([input, pdf_content[0], prompt])
     return json.loads(response.text[8:-4])