Update app.py

app.py

@@ -1,17 +1,17 @@
+from spaces import GPU
 import gradio as gr
 import torch
 from transformers import pipeline
 
-# ✅ Auto-select device: GPU (if available), else CPU
-device = 0 if torch.cuda.is_available() else -1
-
-# ✅ Load text generation pipeline
-pipe = pipeline(
-    "text-generation",
-    model="Pisethan/khmer-lesson-model-v2",
-    tokenizer="Pisethan/khmer-lesson-model",
-    device=device  # Use GPU if available
-)
+@GPU  # 👈 REQUIRED to activate GPU on Hugging Face ZeroGPU
+def main():
+    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
+    )
 
 # ✅ Dropdown choices
 grade_options = ["1", "2", "3", "4", "5", "6"]
@@ -74,3 +74,5 @@ with gr.Blocks() as demo:
 
 # ✅ Launch app with public link
 demo.launch(share=True)
+
+main()