initial commit

Dockerfile

@@ -0,0 +1,16 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,36 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
+app = FastAPI()
+
+# Load tokenizer and model from Hugging Face Hub
+import os
+model_name = os.getenv("MODEL_NAME", "titanabrian/whatsapp-text-classifier-distilbert")
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+model.eval()
+
+# Labels must match your model training
+labels = ["confirmation", "question", "task", "transactional", "unclassified"]
+
+class Message(BaseModel):
+    text: str
+
+@app.get("/")
+def root():
+    return {"message": "WhatsApp Intent Classifier is ready."}
+
+@app.get("/model")
+def get_model():
+    return {"model_name": model_name}
+
+
+@app.post("/predict")
+def predict(msg: Message):
+    inputs = tokenizer(msg.text, return_tensors="pt", truncation=True, padding=True)
+    with torch.no_grad():
+        outputs = model(**inputs)
+        pred_id = outputs.logits.argmax(dim=-1).item()
+        return {"label": labels[pred_id], "label_id": pred_id}

pyproject.toml

@@ -0,0 +1,20 @@
+[project]
+name = "whatsapp-text-classification-api"
+version = "0.1.0"
+description = ""
+authors = [
+    {name = "titanabrian",email = "titanabrian.05@gmail.com"}
+]
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "fastapi (>=0.115.14,<0.116.0)",
+    "pydantic (>=2.11.7,<3.0.0)",
+    "transformers (>=4.53.1,<5.0.0)",
+    "torch (>=2.7.1,<3.0.0)"
+]
+
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"

requirements.txt

@@ -0,0 +1,49 @@
+annotated-types==0.7.0 ; python_version >= "3.11"
+anyio==4.9.0 ; python_version >= "3.11"
+certifi==2025.6.15 ; python_version >= "3.11"
+charset-normalizer==3.4.2 ; python_version >= "3.11"
+colorama==0.4.6 ; python_version >= "3.11" and platform_system == "Windows"
+fastapi==0.115.14 ; python_version >= "3.11"
+filelock==3.18.0 ; python_version >= "3.11"
+fsspec==2025.5.1 ; python_version >= "3.11"
+hf-xet==1.1.5 ; python_version >= "3.11" and (platform_machine == "x86_64" or platform_machine == "amd64" or platform_machine == "arm64" or platform_machine == "aarch64")
+huggingface-hub==0.33.2 ; python_version >= "3.11"
+idna==3.10 ; python_version >= "3.11"
+jinja2==3.1.6 ; python_version >= "3.11"
+markupsafe==3.0.2 ; python_version >= "3.11"
+mpmath==1.3.0 ; python_version >= "3.11"
+networkx==3.5 ; python_version >= "3.11"
+numpy==2.3.1 ; python_version >= "3.11"
+nvidia-cublas-cu12==12.6.4.1 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-cuda-cupti-cu12==12.6.80 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-cuda-nvrtc-cu12==12.6.77 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-cuda-runtime-cu12==12.6.77 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-cudnn-cu12==9.5.1.17 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-cufft-cu12==11.3.0.4 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-cufile-cu12==1.11.1.6 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-curand-cu12==10.3.7.77 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-cusolver-cu12==11.7.1.2 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-cusparse-cu12==12.5.4.2 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-cusparselt-cu12==0.6.3 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-nccl-cu12==2.26.2 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-nvjitlink-cu12==12.6.85 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+nvidia-nvtx-cu12==12.6.77 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+packaging==25.0 ; python_version >= "3.11"
+pydantic-core==2.33.2 ; python_version >= "3.11"
+pydantic==2.11.7 ; python_version >= "3.11"
+pyyaml==6.0.2 ; python_version >= "3.11"
+regex==2024.11.6 ; python_version >= "3.11"
+requests==2.32.4 ; python_version >= "3.11"
+safetensors==0.5.3 ; python_version >= "3.11"
+setuptools==80.9.0 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11" or python_version >= "3.12"
+sniffio==1.3.1 ; python_version >= "3.11"
+starlette==0.46.2 ; python_version >= "3.11"
+sympy==1.14.0 ; python_version >= "3.11"
+tokenizers==0.21.2 ; python_version >= "3.11"
+torch==2.7.1 ; python_version >= "3.11"
+tqdm==4.67.1 ; python_version >= "3.11"
+transformers==4.53.1 ; python_version >= "3.11"
+triton==3.3.1 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.11"
+typing-extensions==4.14.1 ; python_version >= "3.11"
+typing-inspection==0.4.1 ; python_version >= "3.11"
+urllib3==2.5.0 ; python_version >= "3.11"