Uni-Imaginator-v1 Final Push

Removed the redundant comments from the main-code and corrected the model's call.

app.py

@@ -26,13 +26,13 @@ PROMPT_VARIATIONS = [
 ]
 
 def process_variation(variation, input_image, product_name):
-    """Process one variation: call the API, save the generated image to a temporary PNG file, and return its path."""
+    "
     text_input = (
         f"Hi, this is a picture of a product. The name of the product is {product_name}.",
         variation
     )
     response = client.models.generate_content(
-        model=univin_model,  # Use the 'univin' model name from the environment variable
+        model=univin_model,  
         contents=[text_input, input_image],
         config=types.GenerateContentConfig(response_modalities=['Text', 'Image'])
     )
@@ -45,7 +45,7 @@ def process_variation(variation, input_image, product_name):
     return None
 
 def generate_images(input_image, product_name):
-    """Generate images concurrently for the defined prompt variations (exactly 10 calls)."""
+    
     with concurrent.futures.ThreadPoolExecutor() as executor:
         futures = [
             executor.submit(process_variation, variation, input_image, product_name)
@@ -61,7 +61,6 @@ with gr.Blocks() as demo:
         product_name = gr.Textbox(label="Product Name", placeholder="Enter the product name")
     generate_button = gr.Button("Generate Images")
     
-    # The Gallery displays file paths so that each image downloads as a PNG file.
     gallery = gr.Gallery(label="Generated Images", elem_id="gallery", columns=4, height="auto", object_fit="contain", interactive=True)
     
     generate_button.click(fn=generate_images, inputs=[input_image, product_name], outputs=gallery)