OriLib commited on
Commit
d6e753e
1 Parent(s): 3f8e328

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,13 +4,15 @@ import torch.nn.functional as F
4
  from torchvision.transforms.functional import normalize
5
  # from foo import hello
6
  import gradio as gr
 
 
 
7
  # import git # pip install gitpython
8
 
9
  # hello()
10
 
11
  # git.Git(".").clone("https://huggingface.co/briaai/RMBG-1.4")
12
  # git.Git(".").clone("git@hf.co:briaai/RMBG-1.4")
13
- from briarmbg import BriaRMBG
14
 
15
  net=BriaRMBG()
16
  model_path = "./model.pth"
 
4
  from torchvision.transforms.functional import normalize
5
  # from foo import hello
6
  import gradio as gr
7
+ from briarmbg import BriaRMBG
8
+ import PIL
9
+ from PIL import Image
10
  # import git # pip install gitpython
11
 
12
  # hello()
13
 
14
  # git.Git(".").clone("https://huggingface.co/briaai/RMBG-1.4")
15
  # git.Git(".").clone("git@hf.co:briaai/RMBG-1.4")
 
16
 
17
  net=BriaRMBG()
18
  model_path = "./model.pth"