StevenMup2004 commited on
Commit
c692b8b
·
verified ·
1 Parent(s): 7821bac

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -76,7 +76,7 @@ class RouterMLP(nn.Module):
76
  return self.cls(x).squeeze(-1)
77
 
78
  # 2. Download and Load the Checkpoint
79
- repo_id = "YOUR_HF_USERNAME/safe-route-dynaguard" # <-- Replace with your repo name
80
  model_path = hf_hub_download(repo_id=repo_id, filename="model.pt")
81
 
82
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
76
  return self.cls(x).squeeze(-1)
77
 
78
  # 2. Download and Load the Checkpoint
79
+ repo_id = "StevenMup2004/DynaRoute" # <-- Replace with your repo name
80
  model_path = hf_hub_download(repo_id=repo_id, filename="model.pt")
81
 
82
  device = "cuda" if torch.cuda.is_available() else "cpu"