Spaces:
Running
on
Zero
Running
on
Zero
Nan Xue
commited on
Commit
·
98dd04e
1
Parent(s):
21db24d
update
Browse files
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))
|