hbanduk commited on
Commit
9e73d63
·
verified ·
1 Parent(s): 3240e8b

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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(MODEL_REPO)
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")))