Spaces:
Runtime error
Runtime error
max
commited on
Commit
•
aab7fa9
1
Parent(s):
578ad9e
new finetuned model
Browse files- app.py +14 -3
- models/Places_512_FullData+LAION300k+OPM1200k.pkl +3 -0
app.py
CHANGED
|
@@ -237,6 +237,13 @@ class Predictor:
|
|
| 237 |
truncation_psi=1.,
|
| 238 |
noise_mode='const',
|
| 239 |
),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 240 |
}
|
| 241 |
|
| 242 |
# The arguments and types the model takes as input
|
|
@@ -309,8 +316,12 @@ outwithoutalpha = gc.Image(
|
|
| 309 |
mask = gc.Image(label="outpainting mask", type='pil')
|
| 310 |
|
| 311 |
model = gc.Dropdown(
|
| 312 |
-
choices=['places2',
|
| 313 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 314 |
label='model',
|
| 315 |
)
|
| 316 |
|
|
@@ -319,7 +330,7 @@ maturl = 'https://github.com/fenglinglwb/MAT'
|
|
| 319 |
gr.Interface(
|
| 320 |
_outpaint,
|
| 321 |
[searchimage, to_size, border, seed, size, model],
|
| 322 |
-
[outwithoutalpha,out, mask],
|
| 323 |
title=f"MAT Primer for Stable Diffusion\n\nbased on MAT: Mask-Aware Transformer for Large Hole Image Inpainting\n\n{maturl}",
|
| 324 |
description=f"""<html>
|
| 325 |
create an primer for use in stable diffusion outpainting<br>
|
|
|
|
| 237 |
truncation_psi=1.,
|
| 238 |
noise_mode='const',
|
| 239 |
),
|
| 240 |
+
"places2+laion300k+laion1200k(opmasked)": Inpainter(
|
| 241 |
+
network_pkl='models/Places_512_FullData+LAION300k+OPM1200k.pkl',
|
| 242 |
+
resolution=512,
|
| 243 |
+
truncation_psi=1.,
|
| 244 |
+
noise_mode='const',
|
| 245 |
+
),
|
| 246 |
+
|
| 247 |
}
|
| 248 |
|
| 249 |
# The arguments and types the model takes as input
|
|
|
|
| 316 |
mask = gc.Image(label="outpainting mask", type='pil')
|
| 317 |
|
| 318 |
model = gc.Dropdown(
|
| 319 |
+
choices=['places2',
|
| 320 |
+
'places2+laion300k',
|
| 321 |
+
'places2+laion300k+laion300k(opmasked)',
|
| 322 |
+
'places2+laion300k+laion1200k(opmasked)'
|
| 323 |
+
],
|
| 324 |
+
value='places2+laion300k+laion1200k(opmasked)',
|
| 325 |
label='model',
|
| 326 |
)
|
| 327 |
|
|
|
|
| 330 |
gr.Interface(
|
| 331 |
_outpaint,
|
| 332 |
[searchimage, to_size, border, seed, size, model],
|
| 333 |
+
[outwithoutalpha, out, mask],
|
| 334 |
title=f"MAT Primer for Stable Diffusion\n\nbased on MAT: Mask-Aware Transformer for Large Hole Image Inpainting\n\n{maturl}",
|
| 335 |
description=f"""<html>
|
| 336 |
create an primer for use in stable diffusion outpainting<br>
|
models/Places_512_FullData+LAION300k+OPM1200k.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f9ecebfd38f952abd3fde0a74caba64333627a80660f8c14699c1778232231e2
|
| 3 |
+
size 661315824
|