File size: 1,095 Bytes
6f261ab 7a2c85b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
from flask import Flask, jsonify, request, send_file
from flask_cors import CORS
from rembg import remove
from PIL import Image
import io
# Initialize the Flask app
myapp = Flask(__name__)
CORS(myapp) # Enable CORS if needed
@myapp.route('/')
def home():
return "Welcome to the Image Background Remover!" # Basic home response
@myapp.route('/remove_background', methods=['POST'])
def remove_background():
if 'image' not in request.files:
return jsonify({"error": "No image provided"}), 400
input_image = request.files['image'].read() # Read the uploaded image
output_bytes = remove(input_image) # Process the image with rembg
# Convert the output bytes back into a PIL image
output_image = Image.open(io.BytesIO(output_bytes))
img_byte_arr = io.BytesIO()
output_image.save(img_byte_arr, format='PNG')
img_byte_arr.seek(0)
return send_file(img_byte_arr, mimetype='image/png')
# Add this block to make sure your app runs when called
if __name__ == "__main__":
myapp.run(host='0.0.0.0', port=7860) # Run directly if needed for testing |