mithenks commited on
Commit
3b5e294
·
1 Parent(s): f263316

add selectors for model parameters

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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)