Update app.py

app.py

@@ -5,33 +5,29 @@ from PIL import Image
 import io
 
 app = Flask(__name__)
-CORS(app)  # Enable CORS for all routes
+CORS(app)
+
+# Health check endpoint
+@app.route('/health')
+def health_check():
+    return {'status': 'ready'}, 200
 
 @app.route('/remove_bg', methods=['POST'])
 def remove_bg():
-    # Check if image file is present
     if 'file' not in request.files:
         return {'error': 'No file uploaded'}, 400
     
     file = request.files['file']
     
-    # Check if file is an image
-    if file.filename == '':
-        return {'error': 'No selected file'}, 400
-    
     try:
-        # Read image file
         input_image = Image.open(file.stream)
+        # Use specific model and only necessary parameters
+        output_image = remove(input_image, session_new=remove.new_session('u2net'))
         
-        # Remove background
-        output_image = remove(input_image)
-        
-        # Convert to bytes
         img_byte_arr = io.BytesIO()
         output_image.save(img_byte_arr, format='PNG')
         img_byte_arr.seek(0)
         
-        # Return result
         return send_file(img_byte_arr, mimetype='image/png')
     
     except Exception as e: