aixsatoshi commited on
Commit
c5d3fee
·
verified ·
1 Parent(s): 953563e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def mistral_inference(prompt, image_url):
31
  images = encoded.images
32
  tokens = encoded.tokens
33
 
34
- out_tokens, _ = generate([tokens], model, images=[images], max_tokens=256, temperature=0.35, eos_id=tokenizer.instruct_tokenizer.tokenizer.eos_id)
35
  result = tokenizer.decode(out_tokens[0])
36
 
37
  return result
 
31
  images = encoded.images
32
  tokens = encoded.tokens
33
 
34
+ out_tokens, _ = generate([tokens], model, images=[images], max_tokens=1024, temperature=0.35, eos_id=tokenizer.instruct_tokenizer.tokenizer.eos_id)
35
  result = tokenizer.decode(out_tokens[0])
36
 
37
  return result