schirrmacher commited on
Commit
0cdd62a
1 Parent(s): 065cbcc

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. hf_space/app.py +2 -2
hf_space/app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
  from ormbg import ORMBG
7
  from PIL import Image
8
 
9
- model_path = "models/ormbg.pth"
10
 
11
  # Load the model globally but don't send to device yet
12
  net = ORMBG()
@@ -85,4 +85,4 @@ demo = gr.Interface(
85
  )
86
 
87
  if __name__ == "__main__":
88
- demo.launch(share=False, root_path="./", allowed_paths=["hf_space", "models"])
 
6
  from ormbg import ORMBG
7
  from PIL import Image
8
 
9
+ model_path = "./models/ormbg.pth"
10
 
11
  # Load the model globally but don't send to device yet
12
  net = ORMBG()
 
85
  )
86
 
87
  if __name__ == "__main__":
88
+ demo.launch(share=False, root_path="./", allowed_paths=["./hf_space", "./models"])