vikhyatk commited on
Commit
b2d3b41
1 Parent(s): b3a8054

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -51,6 +51,10 @@ def answer_question(img, prompt):
51
  "question": prompt,
52
  "tokenizer": tokenizer,
53
  "streamer": streamer,
 
 
 
 
54
  },
55
  )
56
  thread.start()
 
51
  "question": prompt,
52
  "tokenizer": tokenizer,
53
  "streamer": streamer,
54
+ "repetition_penalty": 1.2,
55
+ "temperature": 0.1,
56
+ "do_sample": True,
57
+ "length_penalty": 1.2
58
  },
59
  )
60
  thread.start()