Update app.py

app.py

@@ -8,11 +8,13 @@ import torch
 
 app = FastAPI()
 
-# Load the RMBG V1.4 model and processor
+# Load the RMBG V1.4 model and processor with trust_remote_code=True
 model = AutoModelForImageSegmentation.from_pretrained(
     "briaai/RMBG-1.4", trust_remote_code=True
 )
-processor = AutoProcessor.from_pretrained("briaai/RMBG-1.4")
+processor = AutoProcessor.from_pretrained(
+    "briaai/RMBG-1.4", trust_remote_code=True
+)
 
 @app.post("/remove-background")
 async def remove_background(file: UploadFile = File(...)):