Update app.py

app.py

@@ -5,8 +5,13 @@ from PIL import Image
 from io import BytesIO
 import tempfile
 import concurrent.futures
+import os  
+
+api_key = os.environ.get("API_KEY")  
+univin_model = os.environ.get("univin")  
+
+client = genai.Client(api_key=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 +32,7 @@ def process_variation(variation, input_image, product_name):
         variation
     )
     response = client.models.generate_content(
-        model=os.environ.getattribute(univin),  
+        model=univin_model,  # Use the 'univin' model name from the environment variable
         contents=[text_input, input_image],
         config=types.GenerateContentConfig(response_modalities=['Text', 'Image'])
     )
@@ -50,7 +55,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")
@@ -61,4 +66,4 @@ with gr.Blocks() as demo:
     
     generate_button.click(fn=generate_images, inputs=[input_image, product_name], outputs=gallery)
     
-demo.launch()
+demo.launch()