Update app.py
Browse files
app.py
CHANGED
|
@@ -75,8 +75,8 @@ if file_name is not None:
|
|
| 75 |
pixel_values.to(device),
|
| 76 |
decoder_input_ids=decoder_input_ids.to(device),
|
| 77 |
max_length=model_ext.decoder.config.max_position_embeddings,
|
| 78 |
-
pad_token_id=
|
| 79 |
-
eos_token_id=
|
| 80 |
use_cache=True,
|
| 81 |
bad_words_ids=[[processor_ext.tokenizer.unk_token_id]],
|
| 82 |
return_dict_in_generate=True,
|
|
|
|
| 75 |
pixel_values.to(device),
|
| 76 |
decoder_input_ids=decoder_input_ids.to(device),
|
| 77 |
max_length=model_ext.decoder.config.max_position_embeddings,
|
| 78 |
+
pad_token_id=processor_ext.tokenizer.pad_token_id,
|
| 79 |
+
eos_token_id=processor_ext.tokenizer.eos_token_id,
|
| 80 |
use_cache=True,
|
| 81 |
bad_words_ids=[[processor_ext.tokenizer.unk_token_id]],
|
| 82 |
return_dict_in_generate=True,
|