mboss commited on
Commit
6ef8dac
·
1 Parent(s): 8ffc5e2

Try to fetch DPT model in advance

Browse files
Files changed (1) hide show
  1. gradio_demo.py +2 -0
gradio_demo.py CHANGED
@@ -8,11 +8,13 @@ from marble import (
8
  run_parametric_control,
9
  setup_control_mlps,
10
  setup_pipeline,
 
11
  )
12
 
13
  # Setup the pipeline and control MLPs
14
  control_mlps = setup_control_mlps()
15
  ip_adapter = setup_pipeline()
 
16
  get_session()
17
 
18
  # Load example images
 
8
  run_parametric_control,
9
  setup_control_mlps,
10
  setup_pipeline,
11
+ get_dpt_model,
12
  )
13
 
14
  # Setup the pipeline and control MLPs
15
  control_mlps = setup_control_mlps()
16
  ip_adapter = setup_pipeline()
17
+ get_dpt_model()
18
  get_session()
19
 
20
  # Load example images