skjaini commited on
Commit
c7602de
·
verified ·
1 Parent(s): 2325a30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,9 +14,9 @@ genai.configure(api_key=GOOGLE_API_KEY)
14
  # Function to generate the modified image
15
  def generate_modified_image(uploaded_image, background_description):
16
  try:
17
- model = genai.GenerativeModel('gemini-pro-vision')
 
18
 
19
- # Open the uploaded file as a PIL Image
20
  img = Image.open(uploaded_image)
21
 
22
  prompt_parts = [
 
14
  # Function to generate the modified image
15
  def generate_modified_image(uploaded_image, background_description):
16
  try:
17
+ # Update the model name here
18
+ model = genai.GenerativeModel('gemini-1.5-flash')
19
 
 
20
  img = Image.open(uploaded_image)
21
 
22
  prompt_parts = [