Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -70,7 +70,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
70 |
|
71 |
model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILENAME)
|
72 |
|
73 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
74 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_REPO)
|
75 |
|
76 |
model.load_state_dict(torch.load(model_path, map_location=torch.device("cpu")))
|
|
|
70 |
|
71 |
model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILENAME)
|
72 |
|
73 |
+
tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', do_lower_case=True)
|
74 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_REPO)
|
75 |
|
76 |
model.load_state_dict(torch.load(model_path, map_location=torch.device("cpu")))
|