Update app.py

app.py

@@ -1,33 +1,34 @@
-import gradio as gr
-from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
-
-# Load model & tokenizer from local directory
-model_path = "./khmer_lesson_model"
-
-model = AutoModelForCausalLM.from_pretrained(model_path, local_files_only=True)
-tokenizer = AutoTokenizer.from_pretrained(model_path, local_files_only=True)
-
-pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
-
-def generate_lesson(grade, topic, level):
-    prompt = f"""Generate a Khmer math lesson plan.
-
-Grade: {grade}
-Topic: {topic}
-TaRL Level: {level}"""
-    output = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)
-    return output[0]["generated_text"]
-
-iface = gr.Interface(
-    fn=generate_lesson,
-    inputs=[
-        gr.Dropdown(["1", "2", "3"], label="ថ្នាក់ (Grade)"),
-        gr.Dropdown(["Addition", "Subtraction", "Counting", "Number Recognition"], label="ប្រធានបទ (Topic)"),
-        gr.Dropdown(["Beginner", "Intermediate", "Advanced"], label="កម្រិត TaRL (TaRL Level)")
-    ],
-    outputs="text",
-    title="អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា (Khmer AI Math Assistant)",
-    description="ជ្រើសរើសថ្នាក់ ប្រធានបទ និងកម្រិតសិស្ស រួចចុចបង្កើតមេរៀន។"
-)
-
-iface.launch()
+import gradio as gr
+from transformers import pipeline
+
+# Load the hosted model and tokenizer from Hugging Face Hub
+pipe = pipeline(
+    "text-generation",
+    model="Pisethan/khmer-lesson-model",
+    tokenizer="Pisethan/khmer-lesson-model"
+)
+
+# Generation function
+def generate_lesson(grade, topic, level):
+    prompt = f"""Generate a Khmer math lesson plan.
+
+Grade: {grade}
+Topic: {topic}
+TaRL Level: {level}"""
+    output = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)
+    return output[0]['generated_text']
+
+# Gradio UI
+iface = gr.Interface(
+    fn=generate_lesson,
+    inputs=[
+        gr.Dropdown(["1", "2", "3", "4", "5", "6"], label="ថ្នាក់ (Grade)"),
+        gr.Dropdown(["Addition", "Subtraction", "Counting", "Number Recognition", "Multiplication", "Division"], label="ប្រធានបទ (Topic)"),
+        gr.Dropdown(["Beginner", "Intermediate", "Advanced"], label="កម្រិតសិស្ស (TaRL Level)")
+    ],
+    outputs=gr.Textbox(label="Khmer Lesson Plan"),
+    title="🤖 អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា",
+    description="ជ្រើសរើសថ្នាក់ ប្រធានបទ និងកម្រិតសិស្ស រួចចុចបង្កើតមេរៀន។ នេះជាម៉ូឌែលកែសម្រួលជាសកលសម្រាប់បង្រៀនគណិតជាភាសាខ្មែរ។"
+)
+
+iface.launch()