Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
2a3b53f
1
Parent(s):
9f1ae40
Added Relight image gen
Browse files
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,
|
526 |
|
527 |
-
|
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]
|