ranaatef commited on
Commit
1a8a9eb
·
verified ·
1 Parent(s): a913d73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def generate_questions(responsibilities: str):
71
  with torch.no_grad():
72
  outputs = model.generate(
73
  **inputs,
74
- max_new_tokens=1024,
75
  temperature=0.7,
76
  do_sample=True
77
  )
 
71
  with torch.no_grad():
72
  outputs = model.generate(
73
  **inputs,
74
+ max_new_tokens=512,
75
  temperature=0.7,
76
  do_sample=True
77
  )