zerogpu support

app.py

@@ -19,7 +19,7 @@ import torch
 import gradio as gr
 from tokenizers import Tokenizer
 from huggingface_hub import hf_hub_download
-
+import spaces
 # ---- point these at your fine-tuned transform model repo -------------------
 # MODEL_REPO = os.environ.get("MODEL_REPO", "uday210/slm-264m-transform")  # <-- EDIT
 # CKPT_FILENAME = os.environ.get("CKPT_FILENAME", "slm-264m-transform.pt")  # <-- EDIT
@@ -64,6 +64,7 @@ EOT_ID = tok.token_to_id(EOT_TOKEN)
 # ---------------------------------------------------------------------------
 # Generation (greedy by default — this task is deterministic transformation)
 # ---------------------------------------------------------------------------
+@spaces.GPU
 @torch.no_grad()
 def generate(prompt: str, max_new_tokens: int = 512, temperature: float = 0.0):
     ids = tok.encode(prompt).ids

requirements.txt

@@ -2,4 +2,5 @@ torch
 tokenizers
 huggingface_hub>=0.25,<1.0
 gradio==5.9.1
-numpy
+numpy
+spaces