Update app.py

app.py

@@ -6,6 +6,9 @@ from google.genai import types
 import gradio as gr
 import io
 from PIL import Image
+from huggingface_hub import login
+
+GEMINI_KEY = os.environ.get("GEMINI_API_KEY")
 
 def save_binary_file(file_name, data):
     f = open(file_name, "wb")
@@ -15,7 +18,7 @@ def save_binary_file(file_name, data):
 def generate_image(prompt, image=None):
     # Initialize client with the API key
     client = genai.Client(
-        api_key="AIzaSyAQcy3LfrkMy6DqS_8MqftAXu1Bx_ov_E8",
+        api_key=GEMINI_KEY,
     )
 
     model = "gemini-2.0-flash-exp-image-generation"