Added Relight image gen

app.py

@@ -522,9 +522,9 @@ def blend_details(input_image, relit_image, masked_image, scaling_factor=1):
     return final_image
 
 @spaces.GPU
-def generate_image(input_img, ai_gen_image, prompt):
+def generate_image(input_img, prompt):
 
-    # ai_gen_image = generate_ai_bg(input_img, prompt)
+    ai_gen_image = generate_ai_bg(input_img, prompt)
 
     mask_input_image = run_grounded_sam(input_img)
 
@@ -542,7 +542,7 @@ def create_ui():
         with gr.Row():
             with gr.Column():
                 input_image = gr.Image(type="pil", label="image")
-                ai_image = gr.Image(type="pil", label="image")
+                # ai_image = gr.Image(type="pil", label="image")
                 prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
                 run_button = gr.Button(value='Run')
 
@@ -554,7 +554,7 @@ def create_ui():
             fn=generate_image, 
             inputs=[
                 input_image,
-                ai_image,
+                # ai_image,
                 prompt
             ], 
             outputs=[output_image]