nikhmr1235 commited on
Commit
f85d437
·
verified ·
1 Parent(s): 4409e88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -157,7 +157,8 @@ def analyze_image_with_gemini(args: dict) -> str:
157
  # Initialize the multimodal LLM (Gemini-Pro-Vision is recommended for image understanding)
158
  # Using a fallback to 'gemini-pro' if 'gemini-pro-vision' isn't directly available or preferred
159
  llm = ChatGoogleGenerativeAI(
160
- model="gemini-pro-vision" if "gemini-pro-vision" in ChatGoogleGenerativeAI.get_available_models(google_api_key) else "gemini-2.0-flash",
 
161
  google_api_key=google_api_key,
162
  temperature=0.0 # Set temperature to 0 for more factual/deterministic responses
163
  )
 
157
  # Initialize the multimodal LLM (Gemini-Pro-Vision is recommended for image understanding)
158
  # Using a fallback to 'gemini-pro' if 'gemini-pro-vision' isn't directly available or preferred
159
  llm = ChatGoogleGenerativeAI(
160
+ #model="gemini-pro-vision" if "gemini-pro-vision" in ChatGoogleGenerativeAI.get_available_models(google_api_key) else "gemini-2.0-flash",
161
+ model="gemini-2.0-flash",
162
  google_api_key=google_api_key,
163
  temperature=0.0 # Set temperature to 0 for more factual/deterministic responses
164
  )