rootglitch commited on
Commit
2a3b53f
·
1 Parent(s): 9f1ae40

Added Relight image gen

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -522,9 +522,9 @@ def blend_details(input_image, relit_image, masked_image, scaling_factor=1):
522
  return final_image
523
 
524
  @spaces.GPU
525
- def generate_image(input_img, ai_gen_image, prompt):
526
 
527
- # ai_gen_image = generate_ai_bg(input_img, prompt)
528
 
529
  mask_input_image = run_grounded_sam(input_img)
530
 
@@ -542,7 +542,7 @@ def create_ui():
542
  with gr.Row():
543
  with gr.Column():
544
  input_image = gr.Image(type="pil", label="image")
545
- ai_image = gr.Image(type="pil", label="image")
546
  prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
547
  run_button = gr.Button(value='Run')
548
 
@@ -554,7 +554,7 @@ def create_ui():
554
  fn=generate_image,
555
  inputs=[
556
  input_image,
557
- ai_image,
558
  prompt
559
  ],
560
  outputs=[output_image]
 
522
  return final_image
523
 
524
  @spaces.GPU
525
+ def generate_image(input_img, prompt):
526
 
527
+ ai_gen_image = generate_ai_bg(input_img, prompt)
528
 
529
  mask_input_image = run_grounded_sam(input_img)
530
 
 
542
  with gr.Row():
543
  with gr.Column():
544
  input_image = gr.Image(type="pil", label="image")
545
+ # ai_image = gr.Image(type="pil", label="image")
546
  prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
547
  run_button = gr.Button(value='Run')
548
 
 
554
  fn=generate_image,
555
  inputs=[
556
  input_image,
557
+ # ai_image,
558
  prompt
559
  ],
560
  outputs=[output_image]