Update app.py

app.py

@@ -34,19 +34,28 @@ if __name__ == "__main__":
     providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
     model_path = huggingface_hub.hf_hub_download("skytnt/anime-seg", "isnetis.onnx")
     rmbg_model = rt.InferenceSession(model_path, providers=providers)
-    app = gr.Blocks()
-    with app:
-        gr.Markdown("# Anime Remove Background\n\n"
-                    "![visitor badge](https://visitor-badge.glitch.me/badge?page_id=skytnt.animeseg)\n\n"
-                    "demo for [https://github.com/SkyTNT/anime-segmentation/](https://github.com/SkyTNT/anime-segmentation/)")
-        with gr.Row():
-            with gr.Column():
-                input_img = gr.Image(label="input image")
-                examples_data = [[f"examples/{x:02d}.jpg"] for x in range(1, 4)]
-                examples = gr.Dataset(components=[input_img], samples=examples_data)
-            run_btn = gr.Button(variant="primary")
-            output_mask = gr.Image(label="mask")
-            output_img = gr.Image(label="result", image_mode="RGBA")
-        examples.click(lambda x: x[0], [examples], [input_img])
-        run_btn.click(rmbg_fn, [input_img], [output_mask, output_img])
-    app.launch(share=True)
+    # app = gr.Blocks()
+    # with app:
+    #     gr.Markdown("# Anime Remove Background\n\n"
+    #                 "![visitor badge](https://visitor-badge.glitch.me/badge?page_id=skytnt.animeseg)\n\n"
+    #                 "demo for [https://github.com/SkyTNT/anime-segmentation/](https://github.com/SkyTNT/anime-segmentation/)")
+    #     with gr.Row():
+    #         with gr.Column():
+    #             input_img = gr.Image(label="input image")
+    #             examples_data = [[f"examples/{x:02d}.jpg"] for x in range(1, 4)]
+    #             examples = gr.Dataset(components=[input_img], samples=examples_data)
+    #         run_btn = gr.Button(variant="primary")
+    #         output_mask = gr.Image(label="mask")
+    #         output_img = gr.Image(label="result", image_mode="RGBA")
+    #     examples.click(lambda x: x[0], [examples], [input_img])
+    #     run_btn.click(rmbg_fn, [input_img], [output_mask, output_img])
+    # app.launch(show_api=True, show_error=True, share=True)
+    gr.Interface(fn=rmbg_fn,
+                    inputs=gr.inputs.Image(label="input image"),    
+                    outputs=[gr.outputs.Image(label="mask", type="numpy", image_mode="L"),
+                            gr.outputs.Image(label="result", type="numpy", image_mode="RGBA")],
+                    title="Anime Remove Background",
+                    description="demo for [https://github.com/SkyTNT/anime-segmentation/](https://github.com/SkyTNT/anime-segmentation/)",
+                    examples=[[f"examples/{x:02d}.jpg"] for x in range(1, 4)],
+                ).launch(show_api=True, show_error=True, share=True)
+