change path

tasks/text.py

@@ -119,7 +119,7 @@ def bert_classifier(test_dataset: dict, model: str):
 def moe_classifier(test_dataset: dict, model: str):
     print("Starting MoE run")
     texts = test_dataset["quote"]
-    model_path = f"theterryzhang/frugal_ai_{model}/0131_MoE_final.pt"
+    model_path = f"tasks/text_models/0131_MoE_final.pt"
 
     embedding_model = AutoModel.from_pretrained("sentence-transformers/all-distilroberta-v1")
     tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-distilroberta-v1")