Update app.py

app.py

@@ -4,6 +4,7 @@ os.environ["GRADIO_ENABLE_SSR"] = "0"
 
 import gradio as gr
 import torch
+import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig
 from accelerate import init_empty_weights, load_checkpoint_and_dispatch
 from datasets import load_dataset
@@ -44,6 +45,7 @@ Few sentences of reasoning
 """
 
 # --- Helper Functions ---
+@spaces.GPU
 def format_rules(rules):
     formatted_rules = "<rules>\n"
     for i, rule in enumerate(rules):