Update app.py

app.py

@@ -25,7 +25,7 @@ def save_binary_file(file_name, data):
     logger.debug(f"File saved successfully: {file_name}")
 
 
-def generate(text, file_name, api_key, model="gemini-1.5-pro-vision"):
+def generate(text, file_name, api_key, model="gemini-1.5-flash"):
     logger.debug(f"Starting generate function with text: '{text}', file_name: '{file_name}', model: '{model}'")
 
     try:
@@ -120,7 +120,7 @@ def process_image_and_prompt(composite_pil, prompt, gemini_api_key):
 
         file_name = composite_path
         input_text = prompt
-        model = "gemini-1.5-pro-vision"  #  Consider changing this to "gemini-pro-vision"
+        model = "gemini-1.5-flash"  #  Consider changing this to "gemini-pro-vision"
 
         gemma_edited_image_path = generate(text=input_text, file_name=file_name, api_key=gemini_api_key, model=model)