Update app.py

app.py

@@ -1,13 +1,15 @@
 import gradio as gr
 from transformers import pipeline
 import torch
+import spaces  # required!
 
-# Dropdown values
+# Define dropdown options
 grade_options = ["1", "2", "3", "4", "5", "6"]
 topic_options = ["Addition", "Subtraction", "Counting", "Number Recognition", "Multiplication", "Division"]
 level_options = ["Beginner", "Intermediate", "Advanced"]
 
-def load_pipeline():
+@spaces.GPU
+def generate_lesson(grade, topic, level):
     device = 0 if torch.cuda.is_available() else -1
     pipe = pipeline(
         "text-generation",
@@ -15,11 +17,7 @@ def load_pipeline():
         tokenizer="Pisethan/khmer-lesson-model",
         device=device
     )
-    return pipe
 
-# Function: generate one
-def generate_lesson(grade, topic, level):
-    pipe = load_pipeline()
     prompt = f"""Generate a Khmer math lesson plan with the following structure:
 
 Lesson Title:
@@ -35,9 +33,16 @@ TaRL Level: {level}"""
     output = pipe(prompt, max_new_tokens=300, temperature=0.7, do_sample=True)
     return output[0]['generated_text']
 
-# Function: generate all combos
+@spaces.GPU
 def generate_all_lessons():
-    pipe = load_pipeline()
+    device = 0 if torch.cuda.is_available() else -1
+    pipe = pipeline(
+        "text-generation",
+        model="Pisethan/khmer-lesson-model-v2",
+        tokenizer="Pisethan/khmer-lesson-model",
+        device=device
+    )
+
     results = ""
     for grade in grade_options:
         for topic in topic_options:
@@ -51,7 +56,7 @@ TaRL Level: {level}"""
                 results += f"🔹 ថ្នាក់ {grade} | {topic} | {level}\n{output[0]['generated_text']}\n\n{'-'*50}\n\n"
     return results
 
-# Gradio UI
+# Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា")
     gr.Markdown("ជ្រើសរើសថ្នាក់ ប្រធានបទ និងកម្រិតសិស្ស រួចចុចបង្កើតមេរៀន។ ឬចុចប៊ូតុងខាងក្រោមសម្រាប់បង្កើតមេរៀនទាំងអស់។")
@@ -72,5 +77,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)
 
-# ❌ Don't use share=True inside HF Space
+demo.queue()  # Required for @spaces.GPU
 demo.launch()