Abaryan commited on
Commit
238b4d9
·
verified ·
1 Parent(s): fa0e902

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -40,9 +40,10 @@ def predict(question: str, option_a: str, option_b: str, option_c: str, option_d
40
  with torch.no_grad():
41
  outputs = model.generate(
42
  **inputs,
43
- max_new_tokens=10,
44
- temperature=0.7,
45
- do_sample=False,
 
46
  pad_token_id=tokenizer.eos_token_id
47
  )
48
 
 
40
  with torch.no_grad():
41
  outputs = model.generate(
42
  **inputs,
43
+ max_new_tokens=256,
44
+ temperature=0.6,
45
+ top_p=0.9,
46
+ do_sample=True,
47
  pad_token_id=tokenizer.eos_token_id
48
  )
49