yasirunet commited on
Commit
0dc4795
·
verified ·
1 Parent(s): f4475fd

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -120,7 +120,7 @@ hard one (risking a bad answer) — the direction you want a router to err in.
120
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
121
  import torch
122
 
123
- repo_id = "appriai/gen-router-tier-classifier"
124
  tokenizer = AutoTokenizer.from_pretrained(repo_id)
125
  model = AutoModelForSequenceClassification.from_pretrained(repo_id)
126
 
@@ -138,7 +138,7 @@ import onnxruntime as ort
138
  from transformers import AutoTokenizer
139
  from huggingface_hub import hf_hub_download
140
 
141
- repo_id = "appriai/gen-router-tier-classifier"
142
  tokenizer = AutoTokenizer.from_pretrained(repo_id)
143
  onnx_path = hf_hub_download(repo_id, "onnx/model.onnx")
144
  session = ort.InferenceSession(onnx_path, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
 
120
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
121
  import torch
122
 
123
+ repo_id = "appriai/gen-router-t1"
124
  tokenizer = AutoTokenizer.from_pretrained(repo_id)
125
  model = AutoModelForSequenceClassification.from_pretrained(repo_id)
126
 
 
138
  from transformers import AutoTokenizer
139
  from huggingface_hub import hf_hub_download
140
 
141
+ repo_id = "appriai/gen-router-t1"
142
  tokenizer = AutoTokenizer.from_pretrained(repo_id)
143
  onnx_path = hf_hub_download(repo_id, "onnx/model.onnx")
144
  session = ort.InferenceSession(onnx_path, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])