Update app.py

app.py

@@ -9,7 +9,6 @@ from dataclasses import dataclass
 from pathlib import Path
 import spaces
 
-@spaces.GPU
 @dataclass
 class SymbolicConfig:
     repo_id: str = "AbstractPhil/bert-beatrix-2048"
@@ -30,6 +29,7 @@ model = AutoModelForMaskedLM.from_pretrained(model_dir).eval().cuda()
 
 MASK_TOKEN = tokenizer.mask_token or "[MASK]"
 
+@spaces.GPU
 def mask_and_predict(text: str, selected_roles: list[str]):
     results = []
     masked_text = text