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
|