Spaces:
Runtime error
Runtime error
chats-bug
commited on
Commit
·
046f505
1
Parent(s):
9d074e1
Trying Attention masks
Browse files
app.py
CHANGED
|
@@ -69,6 +69,7 @@ def generate_caption(
|
|
| 69 |
|
| 70 |
generated_ids = model.generate(
|
| 71 |
pixel_values=pixel_values,
|
|
|
|
| 72 |
max_length=50,
|
| 73 |
)
|
| 74 |
|
|
|
|
| 69 |
|
| 70 |
generated_ids = model.generate(
|
| 71 |
pixel_values=pixel_values,
|
| 72 |
+
attention_mask=inputs.attention_mask,
|
| 73 |
max_length=50,
|
| 74 |
)
|
| 75 |
|