Update app.py

app.py

@@ -40,6 +40,7 @@ def inference(input_img, num_inference_steps, guidance_scale, seed ):
     return result
 
 def remove_background(result):
+    result = Image.open(result)
     result = rembg.remove(result)
     return result
 
@@ -61,7 +62,7 @@ with gr.Blocks() as demo:
     btn = gr.Button('Submit')
 
     btn.click(inference, [input_img, num_inference_steps, guidance_scale, seed ], output_img)
-    rm_in_bkg.input(remove_background, input_img, output_img)
+    rm_in_bkg.input(remove_background, input_img, input_img)
     rm_out_bkg.input(remove_background, output_img, output_img)
     
     gr.Examples(