Ujeshhh commited on
Commit
42675a0
·
verified ·
1 Parent(s): bcb5737

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ genai.configure(api_key="AIzaSyDnx_qUjGTFG1pv1otPUhNt_bGGv14aMDI") # Replace wi
8
  def predict_rat(image):
9
  """Predict if the uploaded image contains a rat using Google Gemini Pro Vision API."""
10
 
11
- model = genai.GenerativeModel("gemini-1.5-flash")
12
 
13
  # Open image using PIL
14
  img = Image.open(image)
 
8
  def predict_rat(image):
9
  """Predict if the uploaded image contains a rat using Google Gemini Pro Vision API."""
10
 
11
+ model = genai.GenerativeModel("learnlm-2.0-flash-experimental")
12
 
13
  # Open image using PIL
14
  img = Image.open(image)