seronk commited on
Commit
3cc5d6a
·
verified ·
1 Parent(s): a70b014

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +2 -2
tasks/text.py CHANGED
@@ -66,8 +66,8 @@ async def evaluate_text(request: TextEvaluationRequest):
66
 
67
 
68
  model_name = "seronk/distillbert-frugal-ai"
69
- model = AutoModelForSequenceClassification.from_pretrained(model_name, local_files_only=True)
70
- tokenizer = AutoTokenizer.from_pretrained(model_name,local_files_only=True)
71
 
72
  def preprocess_function(df):
73
  return tokenizer(df["quote"], truncation=True)
 
66
 
67
 
68
  model_name = "seronk/distillbert-frugal-ai"
69
+ model = AutoModelForSequenceClassification.from_pretrained(model_name)
70
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
71
 
72
  def preprocess_function(df):
73
  return tokenizer(df["quote"], truncation=True)