Update app.py

app.py

@@ -6,7 +6,7 @@ from io import BytesIO
 import tempfile
 import concurrent.futures
 
-client = genai.Client(API_KEY)
+client = genai.Client(os.environ.getattribute(API_KEY))
 PROMPT_VARIATIONS = [
     "Generate a high quality image with a minimalist white background, bright natural light, and an open lid. Keep the product text unchanged.",
     "Generate a high quality image of a beautiful woman who is using the product. The product and its text must remain exactly as in the original image.",
@@ -27,7 +27,7 @@ def process_variation(variation, input_image, product_name):
         variation
     )
     response = client.models.generate_content(
-        model=univin,  # Ensure this is the correct model identifier.
+        model=os.environ.getattribute(univin),  
         contents=[text_input, input_image],
         config=types.GenerateContentConfig(response_modalities=['Text', 'Image'])
     )
@@ -50,7 +50,7 @@ def generate_images(input_image, product_name):
     return output_files
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Uni-Imaginator")
+    gr.Markdown("#Uni-Imaginator")
     with gr.Row():
         input_image = gr.Image(type="pil", label="Upload Image")
         product_name = gr.Textbox(label="Product Name", placeholder="Enter the product name")