Spaces:
Paused
Paused
add selectors for model parameters
Browse files
app.py
CHANGED
@@ -33,11 +33,11 @@ def process_image(set_use_cache, set_return_dict_in_generate, set_early_stopping
|
|
33 |
max_length=model.decoder.config.max_position_embeddings,
|
34 |
pad_token_id=processor.tokenizer.pad_token_id,
|
35 |
eos_token_id=processor.tokenizer.eos_token_id,
|
36 |
-
use_cache=set_use_cache,
|
37 |
bad_words_ids=[[processor.tokenizer.unk_token_id]],
|
38 |
-
return_dict_in_generate=set_return_dict_in_generate,
|
39 |
-
early_stopping=set_early_stopping,
|
40 |
-
output_scores=set_output_scores
|
41 |
)
|
42 |
|
43 |
print(outputs)
|
|
|
33 |
max_length=model.decoder.config.max_position_embeddings,
|
34 |
pad_token_id=processor.tokenizer.pad_token_id,
|
35 |
eos_token_id=processor.tokenizer.eos_token_id,
|
36 |
+
use_cache=set_use_cache=="True",
|
37 |
bad_words_ids=[[processor.tokenizer.unk_token_id]],
|
38 |
+
return_dict_in_generate=set_return_dict_in_generate=="True",
|
39 |
+
early_stopping=set_early_stopping=="True",
|
40 |
+
output_scores=set_output_scores=="True"
|
41 |
)
|
42 |
|
43 |
print(outputs)
|