Update app.py

app.py

@@ -48,32 +48,32 @@ def save_image(image):
     return None
 
 
-slider1 = ImageSlider(label="birefnet", type="pil")
-slider2 = ImageSlider(label="birefnet", type="pil")
-image = gr.Image(label="Upload an image")
-text = gr.Textbox(label="Paste an image URL")
-download_button = gr.Button("Download Image")  # 다운로드 버튼 추가
-output_file = gr.File()  # 파일 출력 추가
-
-chameleon = load_img("butterfly.jpg", output_type="pil")
-
-url = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
-tab1 = gr.Interface(
-    fn, inputs=image, outputs=[slider1, output_file], examples=[chameleon], api_name="image"
-)
-
-tab2 = gr.Interface(fn, inputs=text, outputs=[slider2, output_file], examples=[url], api_name="text")
-
-
-def process_download(image):
-    return save_image(image[0])
-
-
-download_button.click(process_download, inputs=slider1, outputs=output_file)
-
-demo = gr.TabbedInterface(
-    [tab1, tab2], ["image", "text"], title="birefnet for background removal"
-)
+with gr.Blocks() as demo:
+    image = gr.Image(label="Upload an image")
+    text = gr.Textbox(label="Paste an image URL")
+    download_button = gr.Button("Download Image")
+    output_file = gr.File()
+    slider1 = ImageSlider(label="birefnet", type="pil")
+    slider2 = ImageSlider(label="birefnet", type="pil")
+
+    chameleon = load_img("butterfly.jpg", output_type="pil")
+    url = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
+    
+    with gr.Tab("Image Upload"):
+        tab1 = gr.Interface(
+            fn, inputs=image, outputs=[slider1, output_file], examples=[chameleon], api_name="image"
+        )
+
+    with gr.Tab("Image URL"):
+        tab2 = gr.Interface(
+            fn, inputs=text, outputs=[slider2, output_file], examples=[url], api_name="text"
+        )
+
+    def process_download(image):
+        return save_image(image[0])
+
+    download_button.click(process_download, inputs=slider1, outputs=output_file)
 
 if __name__ == "__main__":
     demo.launch()
+