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