tennant commited on
Commit
0f38a6c
1 Parent(s): 71e1a05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -95,7 +95,7 @@ def gr_caption(x):
95
  return np.clip(img, a_min=0., a_max=1.)
96
 
97
  masked, masked_recon, recon, latent = visual_recon(x, model)
98
- caption_from_model = caption(10, latent, model, tokenizer, )
99
 
100
  masked, masked_recon, recon = map(unnorm_pix, (masked, masked_recon, recon))
101
 
 
95
  return np.clip(img, a_min=0., a_max=1.)
96
 
97
  masked, masked_recon, recon, latent = visual_recon(x, model)
98
+ caption_from_model = caption(8, latent, model, tokenizer, )
99
 
100
  masked, masked_recon, recon = map(unnorm_pix, (masked, masked_recon, recon))
101