Update app.py

app.py

@@ -18,7 +18,8 @@ def generate_lesson(grade, topic, level):
         device=device
     )
 
-    prompt = f"""Generate a Khmer math lesson plan with the following structure:
+    prompt = f"""
+You are a lesson planning assistant. Return only one structured Khmer math lesson plan with these fields:
 
 Lesson Title:
 Objective:
@@ -26,11 +27,22 @@ Activity:
 Instruction (Khmer):
 Materials:
 
+Please follow the structure exactly.
+
 Grade: {grade}
 Topic: {topic}
-TaRL Level: {level}"""
+TaRL Level: {level}
+"""
+
+
+    output = pipe(
+    prompt,
+    max_new_tokens=300,
+    temperature=0.7,
+    do_sample=True,
+    eos_token_id=tokenizer.eos_token_id  # helps terminate early
+)
 
-    output = pipe(prompt, max_new_tokens=300, temperature=0.7, do_sample=True)
     return output[0]['generated_text']
 
 @spaces.GPU