Update app.py

app.py

@@ -20,12 +20,17 @@ transform_image = transforms.Compose(
     ]
 )
 
+output_folder = 'output_images'
+if not os.path.exists(output_folder):
+    os.makedirs(output_folder)
+
 def fn(image):
-    image_path = process_file(image)
     im = load_img(image, output_type="pil")
     im = im.convert("RGB")
     origin = im.copy()
     image = process(im)    
+    image_path = os.path.join(output_folder, "no_bg_image.png")
+    image.save(image_path)
     return (image, origin), image_path
 
 @spaces.GPU
@@ -70,7 +75,7 @@ tab3 = gr.Interface(process_file, inputs=image2, outputs=png_file, examples=["gi
 
 
 demo = gr.TabbedInterface(
-    [tab1, tab2, tab3], ["input image", "input url", "png output"], title="RMBG-2.0 for background removal"
+    [tab1, tab2], ["input image", "input url"], title="RMBG-2.0 for background removal"
 )
 
 if __name__ == "__main__":