Spaces:
Running
Running
Update tasks/text.py
Browse files- 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
|
70 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name
|
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)
|