Use the latest APIs + fix dep conflicts

README.md

@@ -5,9 +5,8 @@ emoji: '📊'
 colorFrom: 'red'
 colorTo: 'gray'
 sdk: 'gradio'
-sdk_version: '6.2.0'
+sdk_version: '6.4.0'
 app_file: 'app.py'
-app_build_command: 'pip install -r requirements.txt'
 license: 'agpl-3.0'
 short_description: 'Display the influence of each token on the prediction.'
 models: ['openai/gpt-oss-20b']

app.py

@@ -1,6 +1,5 @@
 import gradio
 import spaces
-import torch.cuda
 
 import psaiops.common.model
 import psaiops.common.tokenizer
@@ -12,8 +11,12 @@ import psaiops.score.residual.app as app
 psaiops.common.model.get_model(name=app.MODEL, device='cpu')
 
 # but do not instantiate unless necessary
-_MODELS = {'cpu': None, 'cuda': None,}
-_TOKENIZERS = {'cpu': None, 'cuda': None,}
+_MODELS = {
+    'cpu': None,
+    'cuda': None,}
+_TOKENIZERS = {
+    'cpu': psaiops.common.tokenizer.get_tokenizer(name=app.MODEL, device='cpu'),
+    'cuda': psaiops.common.tokenizer.get_tokenizer(name=app.MODEL, device='cuda'),}
 
 # LAZY #########################################################################
 

requirements.txt

@@ -1,7 +1,7 @@
-accelerate>=1.10
+accelerate>=1.12
 deformers>=0.0
 gradio>=6.0
-kernels>=0.10
-psaiops>=0.5
-spaces>=0.44
+kernels>=0.11
+psaiops>=0.6
+spaces>=0.47
 triton==3.4