Kr08 commited on
Commit
2579950
·
verified ·
1 Parent(s): 5ac2684

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ def load_qa_model():
48
  "text-generation",
49
  model=model,
50
  tokenizer=tokenizer,
51
- max_new_tokens=256, # Limit generation as needed
52
  )
53
 
54
  return qa_pipeline
 
48
  "text-generation",
49
  model=model,
50
  tokenizer=tokenizer,
51
+ max_new_tokens=1024, # Limit generation as needed
52
  )
53
 
54
  return qa_pipeline