remove files

tasks/text.py

@@ -3,8 +3,8 @@ from datetime import datetime
 from datasets import load_dataset
 from sklearn.metrics import accuracy_score
 import random
-import skops
-from skops.hub import HubModel
+from skops.hub_utils import download
+from skops.io import load
 
 
 from .utils.evaluation import TextEvaluationRequest
@@ -15,8 +15,7 @@ router = APIRouter()
 DESCRIPTION = "Random Baseline"
 ROUTE = "/text"
 
-MODEL_ID = "theterryzhang/frugalai_preproc_tfidf_xgb"
-model = HubModel(model_id=model_id).load()
+MODEL_PATH = "tasks/text_models/xgb_pipeline.skops"
 
 
 @router.post(ROUTE, tags=["Text Task"], 
@@ -65,6 +64,8 @@ async def evaluate_text(request: TextEvaluationRequest):
     
     # Make random predictions (placeholder for actual model inference)
     true_labels = test_dataset["label"]
+
+    model = load(MODEL_PATH)
     predictions = model.predict(test_dataset["text"])
 
     #--------------------------------------------------------------------------------------------

tasks/text_models/.gitattributes

@@ -0,0 +1 @@
+xgb_pipeline.skops filter=lfs diff=lfs merge=lfs -text