Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer
 import torch
-import spaces  # required!
+import spaces  # required for ZeroGPU
 
 # Define dropdown options
 grade_options = ["1", "2", "3", "4", "5", "6"]
@@ -11,10 +11,11 @@ level_options = ["Beginner", "Intermediate", "Advanced"]
 @spaces.GPU
 def generate_lesson(grade, topic, level):
     device = 0 if torch.cuda.is_available() else -1
+    tokenizer = AutoTokenizer.from_pretrained("Pisethan/khmer-lesson-model")
     pipe = pipeline(
         "text-generation",
         model="Pisethan/khmer-lesson-model-v2",
-        tokenizer="Pisethan/khmer-lesson-model",
+        tokenizer=tokenizer,
         device=device
     )
 
@@ -34,24 +35,25 @@ Topic: {topic}
 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
-)
+        prompt,
+        max_new_tokens=300,
+        temperature=0.7,
+        do_sample=True,
+        eos_token_id=tokenizer.eos_token_id
+    )
 
     return output[0]['generated_text']
 
+
 @spaces.GPU
 def generate_all_lessons():
     device = 0 if torch.cuda.is_available() else -1
+    tokenizer = AutoTokenizer.from_pretrained("Pisethan/khmer-lesson-model")
     pipe = pipeline(
         "text-generation",
         model="Pisethan/khmer-lesson-model-v2",
-        tokenizer="Pisethan/khmer-lesson-model",
+        tokenizer=tokenizer,
         device=device
     )
 
@@ -59,15 +61,32 @@ def generate_all_lessons():
     for grade in grade_options:
         for topic in topic_options:
             for level in level_options:
-                prompt = f"""Generate a Khmer math lesson plan.
+                prompt = f"""
+You are a lesson planning assistant. Return only one structured Khmer math lesson plan with these fields:
+
+Lesson Title:
+Objective:
+Activity:
+Instruction (Khmer):
+Materials:
+
+Please follow the structure exactly.
 
 Grade: {grade}
 Topic: {topic}
-TaRL Level: {level}"""
-                output = pipe(prompt, max_new_tokens=200, temperature=0.7, do_sample=True)
+TaRL Level: {level}
+"""
+                output = pipe(
+                    prompt,
+                    max_new_tokens=300,
+                    temperature=0.7,
+                    do_sample=True,
+                    eos_token_id=tokenizer.eos_token_id
+                )
                 results += f"🔹 ថ្នាក់ {grade} | {topic} | {level}\n{output[0]['generated_text']}\n\n{'-'*50}\n\n"
     return results
 
+
 # Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា")
@@ -79,14 +98,12 @@ with gr.Blocks() as demo:
         level = gr.Dropdown(choices=level_options, label="កម្រិតសិស្ស (TaRL Level)", value="Beginner")
 
     output_box = gr.Textbox(
-    label="📘 Khmer Lesson Plan",
-    lines=20,
-    max_lines=200,
-    show_copy_button=True,
-    autoscroll=True
-)
-
-
+        label="📘 Khmer Lesson Plan",
+        lines=20,
+        max_lines=200,
+        show_copy_button=True,
+        autoscroll=True
+    )
 
     with gr.Row():
         gen_btn = gr.Button("✅ បង្កើតមេរៀន")
@@ -97,5 +114,5 @@ with gr.Blocks() as demo:
     gen_all_btn.click(fn=generate_all_lessons, outputs=output_box)
     clear_btn.click(fn=lambda: "", outputs=output_box)
 
-demo.queue()  # Required for @spaces.GPU
+demo.queue()  # Required for ZeroGPU
 demo.launch()