Nan Xue commited on
Commit
98dd04e
·
1 Parent(s): 21db24d
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,9 +7,6 @@ import random
7
  from pathlib import Path
8
  import json
9
 
10
- from scalelsd.ssl.models.detector import ScaleLSD
11
- from scalelsd.base import show, WireframeGraph
12
- from scalelsd.ssl.misc.train_utils import fix_seeds, load_scalelsd_model
13
 
14
  # Title for the Gradio interface
15
  _TITLE = 'Gradio Demo of ScaleLSD for Structured Representation of Images'
@@ -50,6 +47,9 @@ def process_image(
50
  seed=0,
51
  randomize_seed=False
52
  ):
 
 
 
53
  """core processing function for image inference"""
54
  # set random seed
55
  seed = int(randomize_seed_fn(seed, randomize_seed))
 
7
  from pathlib import Path
8
  import json
9
 
 
 
 
10
 
11
  # Title for the Gradio interface
12
  _TITLE = 'Gradio Demo of ScaleLSD for Structured Representation of Images'
 
47
  seed=0,
48
  randomize_seed=False
49
  ):
50
+ from scalelsd.ssl.models.detector import ScaleLSD
51
+ from scalelsd.base import show, WireframeGraph
52
+ from scalelsd.ssl.misc.train_utils import fix_seeds, load_scalelsd_model
53
  """core processing function for image inference"""
54
  # set random seed
55
  seed = int(randomize_seed_fn(seed, randomize_seed))