Spaces:
Runtime error
Runtime error
max
commited on
Commit
·
b2a8658
1
Parent(s):
b6dd358
new model trained with custom outpainting masks
Browse files
app.py
CHANGED
|
@@ -231,6 +231,12 @@ class Predictor:
|
|
| 231 |
truncation_psi=1.,
|
| 232 |
noise_mode='const',
|
| 233 |
),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 234 |
}
|
| 235 |
|
| 236 |
# The arguments and types the model takes as input
|
|
@@ -303,8 +309,8 @@ outwithoutalpha = gc.Image(
|
|
| 303 |
mask = gc.Image(label="outpainting mask", type='pil')
|
| 304 |
|
| 305 |
model = gc.Dropdown(
|
| 306 |
-
choices=['places2', 'places2+laion300k'],
|
| 307 |
-
value='places2',
|
| 308 |
label='model',
|
| 309 |
)
|
| 310 |
|
|
|
|
| 231 |
truncation_psi=1.,
|
| 232 |
noise_mode='const',
|
| 233 |
),
|
| 234 |
+
"places2+laion300k+laion300k(opmasked)": Inpainter(
|
| 235 |
+
network_pkl='models/Places_512_FullData+LAION300k+OPM300k.pkl',
|
| 236 |
+
resolution=512,
|
| 237 |
+
truncation_psi=1.,
|
| 238 |
+
noise_mode='const',
|
| 239 |
+
),
|
| 240 |
}
|
| 241 |
|
| 242 |
# The arguments and types the model takes as input
|
|
|
|
| 309 |
mask = gc.Image(label="outpainting mask", type='pil')
|
| 310 |
|
| 311 |
model = gc.Dropdown(
|
| 312 |
+
choices=['places2', 'places2+laion300k', 'places2+laion300k+laion300k(opmasked)'],
|
| 313 |
+
value='places2+laion300k+laion300k(opmasked)',
|
| 314 |
label='model',
|
| 315 |
)
|
| 316 |
|
models/Places_512_FullData+LAION300k+OPM300k.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2d2ed6751e2ed8a2120864fd5c1f08a8e926a2f79d5aa91bb35f9cc32869e77f
|
| 3 |
+
size 661315824
|