4

app.py

@@ -29,7 +29,7 @@ def fn(image):
     
     name_path = "output.png"
     processed_image.save(name_path)
-    return (processed_image, name_path)
+    return (origin, name_path)
 
 @spaces.GPU
 def process(image):
@@ -58,18 +58,20 @@ image_upload = gr.Image(label="Upload an image")
 image_file_upload = gr.Image(label="Upload an image", type="filepath")
 url_input = gr.Textbox(label="Paste an image URL")
 output_file = gr.File(label="Output PNG File")
-
+download_button = gr.File(label="Download")
 
 # Example images
 chameleon = load_img("butterfly.jpg", output_type="pil")
 url_example = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
 
-tab1 = gr.Interface(fn, inputs=image_upload, outputs= slider1, examples=[chameleon], api_name="image")
+tab1 = gr.Interface(fn, inputs=image_upload, outputs= [slider1 , download_button], examples=[chameleon], api_name="image")
 tab2 = gr.Interface(fn, inputs=url_input, outputs=slider2, examples=[url_example], api_name="text")
-tab3 = gr.Interface(process_file, inputs=image_file_upload, outputs=output_file, examples=["butterfly.jpg"], api_name="png")
+#tab3 = gr.Interface(process_file, inputs=image_file_upload, outputs=output_file, examples=["butterfly.jpg"], api_name="png")
 
 demo = gr.TabbedInterface(
-    [tab1, tab2, tab3], ["Image Upload", "URL Input", "File Output"], title="Background Removal Tool"
+    #[tab1, tab2, tab3], ["Image Upload", "URL Input", "File Output"], title="Background Removal Tool"
+    [tab1, tab2], ["Image Input", "URL Input"], title="Background Removal Pro"
+
 )
 
 if __name__ == "__main__":