try cuda

app.py

@@ -16,10 +16,10 @@ huggingface_hub.login(token=hf_key)
 tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase-3b")
 vardecoder_model = AutoModelForCausalLM.from_pretrained(
     "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16, device_map={"": 0}
-)
+).to("cuda")
 fielddecoder_model = AutoModelForCausalLM.from_pretrained(
     "ejschwartz/resym-fielddecoder", torch_dtype=torch.bfloat16, device_map={"": 0}
-)
+).to("cuda")
 
 gradio_client = Client("https://ejschwartz-resym-field-helper.hf.space/")