Update app.py

app.py

@@ -1,13 +1,20 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load model from Hugging Face Hub
+
+
+import torch
+
+device = 0 if torch.cuda.is_available() else -1
+
 pipe = pipeline(
     "text-generation",
     model="Pisethan/khmer-lesson-model-v2",
-    tokenizer="Pisethan/khmer-lesson-model"
+    tokenizer="Pisethan/khmer-lesson-model",
+    device=device  # 🔥 use GPU if available
 )
 
+
 # Define dropdown options
 grade_options = ["1", "2", "3", "4", "5", "6"]
 topic_options = ["Addition", "Subtraction", "Counting", "Number Recognition", "Multiplication", "Division"]