tennant commited on
Commit
0a876bf
1 Parent(s): 1ceff34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -101,7 +101,7 @@ def gr_caption(x):
101
  return np.clip(img, a_min=0., a_max=1.)
102
 
103
  masked, masked_recon, recon, latent = visual_recon(x, model)
104
- caption_from_model = caption(20, latent, model, tokenizer, )
105
 
106
  masked, masked_recon, recon = map(unnorm_pix, (masked, masked_recon, recon))
107
  return_img = np.concatenate([masked, masked_recon, recon], axis=1)
 
101
  return np.clip(img, a_min=0., a_max=1.)
102
 
103
  masked, masked_recon, recon, latent = visual_recon(x, model)
104
+ caption_from_model = caption(20, latent, model, tokenizer, prefix='a')
105
 
106
  masked, masked_recon, recon = map(unnorm_pix, (masked, masked_recon, recon))
107
  return_img = np.concatenate([masked, masked_recon, recon], axis=1)