Athspi commited on
Commit
8489828
·
verified ·
1 Parent(s): f503772

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def upload_image(image_data_url):
37
 
38
  def is_prohibited_request(uploaded_file, object_type):
39
  """Check if request is to remove person/animal using gemini-2.0-flash-lite"""
40
- model = "gemini-2.0-flash-lite"
41
  parts = [
42
  types.Part.from_uri(file_uri=uploaded_file.uri, mime_type=uploaded_file.mime_type),
43
  types.Part.from_text(text=f"Remove {object_type}")
 
37
 
38
  def is_prohibited_request(uploaded_file, object_type):
39
  """Check if request is to remove person/animal using gemini-2.0-flash-lite"""
40
+ model = "gemini-2.0-pro-exp-02-05"
41
  parts = [
42
  types.Part.from_uri(file_uri=uploaded_file.uri, mime_type=uploaded_file.mime_type),
43
  types.Part.from_text(text=f"Remove {object_type}")