Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
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 |
|