briaai

/

RMBG-1.4

like 1.84k

Follow

BRIA AI 10k

Description

This PR fixes runtime errors in the example code of the model card.

Changes

• Added missing imports:
  import numpy as np
  import torch
  import torch.nn.functional as F
  import requests
  from io import BytesIO

• Defined missing variable:
  model_input_size = [1024, 1024]

• Replaced Image.open(image_path) with code that properly loads images from a URL:
  response = requests.get(image_path)
  orig_image = Image.open(BytesIO(response.content))

Testing

The code has been successfully tested and runs without error.

Note

This contribution is part of an ongoing research initiative to systematically identify and correct faulty example code in Hugging Face Model Cards.
We would appreciate a timely review and integration of this patch to support code reliability and enhance reproducibility for downstream users.