Update app.py
Browse files
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=
|
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
|