update

tasks/text.py

@@ -72,8 +72,8 @@ async def evaluate_text(request: TextEvaluationRequest):
     ]
 
     
-    tokenizer = AutoTokenizer.from_pretrained("cococli/bert-base-uncased-frugalai")
-    model = AutoModelForSequenceClassification.from_pretrained("cococli/bert-base-uncased-frugalai").to(device)
+    tokenizer = AutoTokenizer.from_pretrained(model_name[0])
+    model = AutoModelForSequenceClassification.from_pretrained(model[0]).to(device)
     
     # def tokenize_function(examples):
     #     return tokenizer(examples["quote"], padding=True, truncation=True, return_tensors='pt')