raksama19 commited on
Commit
8d25dfa
·
verified ·
1 Parent(s): 8892f07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -250,7 +250,9 @@ Provide a descriptive alt text in 1-2 sentences that is informative but not over
250
  max_new_tokens=1024,
251
  disable_compile=True,
252
  do_sample=True,
253
- temperature=0.7
 
 
254
  )
255
 
256
  text = self.processor.batch_decode(
 
250
  max_new_tokens=1024,
251
  disable_compile=True,
252
  do_sample=True,
253
+ temperature=0.7,
254
+ top_p=0.9,
255
+ pad_token_id=self.processor.tokenizer.pad_token_id
256
  )
257
 
258
  text = self.processor.batch_decode(