Add API Route

main.py

@@ -1,6 +1,9 @@
 import gradio as gr
 from rembg import remove
 from PIL import Image
+from flask import Flask, request, jsonify
+
+app = Flask(__name__)
 
 def remove_background(input_image):
     output_image = remove(input_image)
@@ -9,4 +12,13 @@ def remove_background(input_image):
 inputs = gr.inputs.Image()
 outputs = gr.outputs.Image(type='pil')
 
-gr.Interface(fn=remove_background, inputs=inputs, outputs=outputs).launch()
+gr.Interface(fn=remove_background, inputs=inputs, outputs=outputs).launch()
+
+@app.route("/api/remove-background", methods=["POST"])
+def remove_background_api():
+    if request.method == "GET":
+        return "This API removes the background from an image. To use it, send a POST request to this URL with an image file in the 'input' field of the request body."
+
+    input_image = Image.open(request.files["input"])
+    output_image = remove_background(input_image)
+    return jsonify({"output": output_image})