Update app.py

app.py

@@ -614,9 +614,9 @@ def process_bg(input_fg, input_bg, prompt, image_width, image_height, num_sample
 
 @torch.inference_mode()
 def process_relight(input_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source):
-    input_fg, matting = run_rmbg(input_fg)
+    #input_fg, matting = run_rmbg(input_fg)
     results = process(input_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source)
-    return input_fg, results
+    return results
 
 
 
@@ -1142,7 +1142,7 @@ with block:
             #     run_on_click=True, examples_per_page=1024
             # )
         ips = [input_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source]
-        relight_button.click(fn=process_relight, inputs=ips, outputs=[extracted_fg, result_gallery])
+        relight_button.click(fn=process_relight, inputs=ips, outputs=[result_gallery])
         example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
         example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
         find_objects_button.click(