HarisVasilo commited on
Commit
977750d
·
verified ·
1 Parent(s): b9f4a34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -169,8 +169,8 @@ if pdf:
169
  inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
170
  output = model.generate(
171
  **inputs,
172
- max_new_tokens=250,
173
- temperature=0.2,
174
  repetition_penalty=1.2,
175
  eos_token_id=tokenizer.eos_token_id,
176
  )
 
169
  inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
170
  output = model.generate(
171
  **inputs,
172
+ max_new_tokens=500,
173
+ temperature=0.3,
174
  repetition_penalty=1.2,
175
  eos_token_id=tokenizer.eos_token_id,
176
  )