RMBG

Runtime error

OriLib commited on Jan 30

Commit

fe1af7d

•

1 Parent(s): c599dcc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import numpy as np
 import torch.nn.functional as F
 from torchvision.transforms.functional import normalize
-from models import BriaRMBG
 import gradio as gr
 import git  # pip install gitpython

 import numpy as np
 import torch.nn.functional as F
 from torchvision.transforms.functional import normalize
+from briarmbg import BriaRMBG
 import gradio as gr
 import git  # pip install gitpython