Spaces:
Running
on
A10G
Running
on
A10G
Andranik Sargsyan
commited on
Commit
β’
29ac50c
1
Parent(s):
da1e12f
update examples
Browse files- app.py +10 -9
- assets/examples/masked/a19.png +0 -3
- assets/examples/masked/a2.png +0 -3
- assets/examples/masked/a4.png +0 -3
- assets/examples/masked/a40.png +0 -3
- assets/examples/masked/a46.png +0 -3
- assets/examples/masked/a51.png +0 -3
- assets/examples/masked/a54.png +0 -3
- assets/examples/masked/a65.png +0 -3
- assets/examples/sbs/a19.png +2 -2
- assets/examples/sbs/a2.png +2 -2
- assets/examples/sbs/a4.png +2 -2
- assets/examples/sbs/a40.png +2 -2
- assets/examples/sbs/a46.png +2 -2
- assets/examples/sbs/a51.png +2 -2
- assets/examples/sbs/a54.png +2 -2
- assets/examples/sbs/a65.png +2 -2
- lib/methods/sr.py +1 -1
app.py
CHANGED
@@ -32,19 +32,20 @@ example_inputs = [
|
|
32 |
['assets/examples/images/a65.jpg', 'hoodie'],
|
33 |
['assets/examples/images/a54.jpg', 'salad'],
|
34 |
['assets/examples/images/a51.jpg', 'space helmet'],
|
35 |
-
['assets/examples/images/a46.jpg', '
|
36 |
['assets/examples/images/a19.jpg', 'antique greek vase'],
|
37 |
['assets/examples/images/a2.jpg', 'sunglasses'],
|
38 |
]
|
|
|
39 |
thumbnails = [
|
40 |
-
'
|
41 |
-
'
|
42 |
-
'
|
43 |
-
'
|
44 |
-
'
|
45 |
-
'
|
46 |
-
'
|
47 |
-
'
|
48 |
]
|
49 |
|
50 |
example_previews = [
|
|
|
32 |
['assets/examples/images/a65.jpg', 'hoodie'],
|
33 |
['assets/examples/images/a54.jpg', 'salad'],
|
34 |
['assets/examples/images/a51.jpg', 'space helmet'],
|
35 |
+
['assets/examples/images/a46.jpg', 'stack of books'],
|
36 |
['assets/examples/images/a19.jpg', 'antique greek vase'],
|
37 |
['assets/examples/images/a2.jpg', 'sunglasses'],
|
38 |
]
|
39 |
+
|
40 |
thumbnails = [
|
41 |
+
'assets/examples/sbs/a40.png',
|
42 |
+
'assets/examples/sbs/a4.png',
|
43 |
+
'assets/examples/sbs/a65.png',
|
44 |
+
'assets/examples/sbs/a54.png',
|
45 |
+
'assets/examples/sbs/a51.png',
|
46 |
+
'assets/examples/sbs/a46.png',
|
47 |
+
'assets/examples/sbs/a19.png',
|
48 |
+
'assets/examples/sbs/a2.png'
|
49 |
]
|
50 |
|
51 |
example_previews = [
|
assets/examples/masked/a19.png
DELETED
Git LFS Details
|
assets/examples/masked/a2.png
DELETED
Git LFS Details
|
assets/examples/masked/a4.png
DELETED
Git LFS Details
|
assets/examples/masked/a40.png
DELETED
Git LFS Details
|
assets/examples/masked/a46.png
DELETED
Git LFS Details
|
assets/examples/masked/a51.png
DELETED
Git LFS Details
|
assets/examples/masked/a54.png
DELETED
Git LFS Details
|
assets/examples/masked/a65.png
DELETED
Git LFS Details
|
assets/examples/sbs/a19.png
CHANGED
Git LFS Details
|
Git LFS Details
|
assets/examples/sbs/a2.png
CHANGED
Git LFS Details
|
Git LFS Details
|
assets/examples/sbs/a4.png
CHANGED
Git LFS Details
|
Git LFS Details
|
assets/examples/sbs/a40.png
CHANGED
Git LFS Details
|
Git LFS Details
|
assets/examples/sbs/a46.png
CHANGED
Git LFS Details
|
Git LFS Details
|
assets/examples/sbs/a51.png
CHANGED
Git LFS Details
|
Git LFS Details
|
assets/examples/sbs/a54.png
CHANGED
Git LFS Details
|
Git LFS Details
|
assets/examples/sbs/a65.png
CHANGED
Git LFS Details
|
Git LFS Details
|
lib/methods/sr.py
CHANGED
@@ -59,7 +59,7 @@ def refine_mask(hr_image, hr_mask, lr_image, sam_predictor):
|
|
59 |
|
60 |
def run(ddim, sam_predictor, lr_image, hr_image, hr_mask, prompt = 'high resolution professional photo', noise_level=20,
|
61 |
blend_output = True, blend_trick = True, no_superres = False,
|
62 |
-
dt =
|
63 |
torch.manual_seed(seed)
|
64 |
dtype = ddim.vae.encoder.conv_in.weight.dtype
|
65 |
device = ddim.vae.encoder.conv_in.weight.device
|
|
|
59 |
|
60 |
def run(ddim, sam_predictor, lr_image, hr_image, hr_mask, prompt = 'high resolution professional photo', noise_level=20,
|
61 |
blend_output = True, blend_trick = True, no_superres = False,
|
62 |
+
dt = 50, seed = 1, guidance_scale = 7.5, negative_prompt = '', use_sam_mask = False):
|
63 |
torch.manual_seed(seed)
|
64 |
dtype = ddim.vae.encoder.conv_in.weight.dtype
|
65 |
device = ddim.vae.encoder.conv_in.weight.device
|