Update app.py

app.py

@@ -14,13 +14,13 @@ def configure_api_key(api_key):
             raise ValueError("No API key provided and GEMINI_API_KEY environment variable not set.")
         return key
 
-def generate_image_from_text(prompt, api_key, model_name="gemini-1.5-pro-002"): #Or using "gemini-1.0-pro-vision-001" or "gemini-pro"
+def generate_image_from_text(prompt, api_key, model_name="gemini-1.5-pro-002"):  # Corrected model name
     """Generates an image from a text prompt using a specified Gemini model.
 
     Args:
         prompt: The text prompt describing the desired image.
         api_key: Your Google AI API key.
-        model_name: The name of the Gemini model to use (default: gemini-1.5-pro).
+        model_name: The name of the Gemini model to use (default: gemini-1.5-pro-002).
 
     Returns:
         A PIL.Image object representing the generated image, or None on error.
@@ -48,7 +48,7 @@ def generate_image_from_text(prompt, api_key, model_name="gemini-1.5-pro-002"):
         print(f"An error occurred during image generation: {e}")
         return None
 
-def edit_image(image_path, prompt, output_path, api_key, model_name="gemini-1.5-pro-002"): # Or "gemini-1.0-pro-vision-001"
+def edit_image(image_path, prompt, output_path, api_key, model_name="gemini-1.5-pro-002"):  # Corrected model name
     """Edits an image using a specified Gemini model and a text prompt.
 
     Args:
@@ -143,8 +143,8 @@ with gr.Blocks() as demo:
     examples = [
         [None, "A futuristic cityscape at night with flying cars"],
         [None, "A cat wearing a tiny hat"],
-        ["data/1.webp", 'change text to "AMEER"'],
-        ["data/2.webp", "remove the spoon from hand only"],
+        ["data/1.webp", 'change text to "AMEER"'],  # You'll need example images
+        ["data/2.webp", "remove the spoon from hand only"], #  in a 'data' folder
     ] #Combined both image editing and image generation examples.
 
     submit_btn.click(
@@ -159,4 +159,4 @@ with gr.Blocks() as demo:
         label="Try these examples"
     )
 
-demo.launch(share=True)
+demo.launch(share=True)