Spaces:
Sleeping
Sleeping
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)
|