schirrmacher commited on
Commit
5932f9f
1 Parent(s): 48515b1

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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,6 +85,4 @@ demo = gr.Interface(
85
  )
86
 
87
  if __name__ == "__main__":
88
- demo.launch(
89
- share=False, root_path="../", allowed_paths=["../hf_space", "../models"]
90
- )
 
6
  from ormbg import ORMBG
7
  from PIL import Image
8
 
9
+ model_path = "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, allowed_paths=["./"])