Pisethan commited on
Commit
f2b46c6
·
verified ·
1 Parent(s): ce4e9ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def generate_lesson(grade, topic, level):
20
  Grade: {grade}
21
  Topic: {topic}
22
  TaRL Level: {level}"""
23
- output = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)
24
  return output[0]['generated_text']
25
 
26
  # Function: auto generate all combinations
 
20
  Grade: {grade}
21
  Topic: {topic}
22
  TaRL Level: {level}"""
23
+ output = pipe(prompt, max_new_tokens=300, temperature=0.7, do_sample=True)
24
  return output[0]['generated_text']
25
 
26
  # Function: auto generate all combinations