poltextlab commited on
Commit
bdb2bce
·
verified ·
1 Parent(s): b910590

revert token

Browse files
Files changed (1) hide show
  1. interfaces/illframes.py +1 -1
interfaces/illframes.py CHANGED
@@ -34,7 +34,7 @@ def build_huggingface_path(domain: str):
34
 
35
  def predict(text, model_id, tokenizer_id, label_names):
36
  device = torch.device("cpu")
37
- model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto", offload_folder="offload", token=HF_TOKEN, use_auth_token=HF_TOKEN)
38
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
39
 
40
  inputs = tokenizer(text,
 
34
 
35
  def predict(text, model_id, tokenizer_id, label_names):
36
  device = torch.device("cpu")
37
+ model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto", offload_folder="offload", token=HF_TOKEN)
38
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
39
 
40
  inputs = tokenizer(text,