Upload folder using huggingface_hub

.gitignore

@@ -1 +0,0 @@
-model/

main/helper.py

@@ -1,17 +1,8 @@
 # Helper functions for the model inference api
-import os
+
 import yaml
 import joblib
 from pathlib import Path
-from huggingface_hub import snapshot_download
-from dotenv import load_dotenv
-load_dotenv()
-
-# get environment variable
-hf_repo_id = os.getenv("HUGGINGFACE_REPO_ID")
-if not hf_repo_id:
-    raise EnvironmentError("HUGGINGFACE_REPO_ID is not set")
-
 
 # load yaml files to get model meta data.
 try:
@@ -24,17 +15,8 @@ except:
 def load_model():
     """ Loads ML model from location path and returns the model. """
     try:
-        model_dir = snapshot_download(
-            repo_id = str(hf_repo_id),
-            repo_type = "model",
-            local_dir = "/tmp/model",
-            local_dir_use_symlinks = False
-        )
-
-        model_path = Path(model_dir)/"python_model.pkl"
-        with open(model_path, "rb") as f:
-            model = joblib.load(f)
-        
+        with open(Path("model/python_model.pkl"), "rb") as f:
+            model = joblib.load(f)  
         return model
     
     except Exception as e:

model/MLmodel

@@ -0,0 +1,25 @@
+artifact_path: XGB-v2
+flavors:
+  python_function:
+    artifacts:
+      classifier:
+        path: artifacts\XGB-v2.joblib
+        uri: models\XGB-v2.joblib
+      vectorizer:
+        path: artifacts\Tfidf.joblib
+        uri: models\Tfidf.joblib
+    cloudpickle_version: 3.1.1
+    code: null
+    env:
+      conda: conda.yaml
+      virtualenv: python_env.yaml
+    loader_module: mlflow.pyfunc.model
+    python_model: python_model.pkl
+    python_version: 3.11.5
+    streamable: false
+mlflow_version: 2.22.1
+model_size_bytes: 11990188
+model_uuid: 65490db310744bdf8f1c897d96f8aca8
+prompts: null
+run_id: cda6d2d206b34409a74cd67407bda91c
+utc_time_created: '2025-07-28 10:17:07.559763'

model/artifacts/Tfidf.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d3b128625a5b8b778ee4d4a97f8afdfba1268a3ee14b9e3328bab3de48e685cf
+size 120443

model/artifacts/XGB-v2.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa4330bca1029dc4532a5c4ced95b6fa62ef196f6789fad05a1414d662967fea
+size 5863647

model/conda.yaml

@@ -0,0 +1,15 @@
+channels:
+- conda-forge
+dependencies:
+- python=3.11.5
+- pip<=25.1
+- pip:
+  - mlflow==2.22.1
+  - cloudpickle==3.1.1
+  - numpy==2.2.6
+  - pandas==2.3.1
+  - psutil==7.0.0
+  - scikit-learn==1.7.0
+  - scipy==1.13.1
+  - xgboost==3.0.2
+name: mlflow-env

model/python_env.yaml

@@ -0,0 +1,7 @@
+python: 3.11.5
+build_dependencies:
+- pip==25.1
+- setuptools==78.1.1
+- wheel==0.45.1
+dependencies:
+- -r requirements.txt

model/python_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6d00d6029ae727539c833a6504499aee7c3d7da5de56b03be330806293f3954
+size 6006098

model/registered_model_meta

@@ -0,0 +1,2 @@
+model_name: ToxicTagger-Models
+model_version: '6'

model/requirements.txt

@@ -0,0 +1,8 @@
+mlflow==2.22.1
+cloudpickle==3.1.1
+numpy==2.2.6
+pandas==2.3.1
+psutil==7.0.0
+scikit-learn==1.7.0
+scipy==1.13.1
+xgboost==3.0.2

requirements.txt

@@ -1,5 +1,4 @@
 fastapi==0.116.1
 uvicorn==0.35.0
 joblib==1.5.1
-PyYAML==6.0.2
-python-dotenv==1.1.1
+PyYAML==6.0.2