Spaces:
Running
Running
revert token
Browse files- 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
|
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,
|