visheratin commited on
Commit
588861c
1 Parent(s): 3aabbc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def answer_question(image, question, max_crops, num_tokens, sample, temperature,
61
  "eos_token_id": processor.tokenizer.eos_token_id,
62
  "pad_token_id": processor.tokenizer.eos_token_id,
63
  "temperature": temperature,
64
- "sample": sample,
65
  "top_k": top_k,
66
  }
67
  thread = Thread(target=model.generate, kwargs=generation_kwargs)
 
61
  "eos_token_id": processor.tokenizer.eos_token_id,
62
  "pad_token_id": processor.tokenizer.eos_token_id,
63
  "temperature": temperature,
64
+ "do_sample": sample,
65
  "top_k": top_k,
66
  }
67
  thread = Thread(target=model.generate, kwargs=generation_kwargs)