switch to pyvene

app.py

@@ -72,7 +72,7 @@ if torch.cuda.is_available():
     path_to_md = hf_hub_download(repo_id="pyvene/gemma-reft-2b-it-res", filename="l20/metadata.jsonl", force_download=False)
     params = torch.load(path_to_params).cuda()
     md = load_jsonl(path_to_md)
-    id_to_concept = {item["id"]: item["concept"] for item in md}
+    id_to_concept = {item["concept_id"]: item["concept"] for item in md}
     concept_list = [item["concept"] for item in md]
 
     steer = Steer(embed_dim=params.shape[0], latent_dim=params.shape[1])