Update tasks/text.py

tasks/text.py

@@ -106,9 +106,9 @@ def bert_classifier(test_dataset: dict, model: str):
 
     tokenizer = AutoTokenizer.from_pretrained(model_repo)
 
-    if model.isin(['bert_base_pruned']):
+    if model in ['bert_base_pruned']:
         model = AutoModelForSequenceClassification.from_pretrained(model_repo)
-    elif model.isin(['sbert_distilroberta']):
+    elif model in ['sbert_distilroberta']:
         model = SentenceBERTMultiClass.from_pretrained(model_repo)
     else:
         raise(ValueError)