mlbench123 commited on
Commit
c1657df
·
verified ·
1 Parent(s): 2fb26de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def remove_bg(image: np.ndarray) -> np.ndarray:
42
  image = Image.fromarray(image)
43
  input_images = transform_image(image).unsqueeze(0).to("cpu")
44
 
45
- , "en" # Prediction
46
  with torch.no_grad():
47
  preds = birefnet(input_images)[-1].sigmoid().cpu()
48
  pred = preds[0].squeeze()
 
42
  image = Image.fromarray(image)
43
  input_images = transform_image(image).unsqueeze(0).to("cpu")
44
 
45
+ # Prediction
46
  with torch.no_grad():
47
  preds = birefnet(input_images)[-1].sigmoid().cpu()
48
  pred = preds[0].squeeze()