Upload 5 files

.gitattributes

@@ -1,35 +1,8 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.py text eol=lf
+*.md text eol=lf
+*.txt text eol=lf
+*.yml text eol=lf
+*.yaml text eol=lf
+*.json text eol=lf
+Dockerfile text eol=lf
+.gitignore text eol=lf

.gitignore

@@ -1,20 +1,14 @@
-# environment
 venv/
-.venv/
-
-# ML / MLflow
-mlflow/
-mlartifacts/
-*.keras
-*.h5
-*.pkl
-
-
-
-# Python cache
 __pycache__/
 *.pyc
-
-# IDE
-.vscode/
-.idea/
+.DS_Store
+storage/images/*
+storage/segments/*
+storage/results/*
+!storage/images/.gitkeep
+!storage/segments/.gitkeep
+!storage/results/.gitkeep
+models/
+*.h5
+*.keras
+.env

Dockerfile

@@ -1,19 +1,12 @@
-FROM python:3.10
+FROM python:3.11-slim
 
 WORKDIR /app
-#
-# 🔥 Set PYTHONPATH to include root directory (for sitecustomize.py)
-ENV PYTHONPATH=/app:${PYTHONPATH}
 
-# install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# copy project files
 COPY . .
 
-# expose port (Hugging Face uses 7860)
 EXPOSE 7860
 
-# run app
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -1,12 +1,42 @@
 ---
-title: Api
-emoji: 🏆
-colorFrom: yellow
-colorTo: pink
+title: Skin Cancer Detection API
+emoji: 🏥
+colorFrom: blue
+colorTo: red
 sdk: docker
 pinned: false
 license: mit
-short_description: fast api for models
+short_description: FastAPI for skin cancer classification & segmentation
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Skin Cancer Detection API
+
+FastAPI application for skin cancer classification and segmentation, deployed on Hugging Face Spaces.
+
+## API Endpoints
+
+| Method | Endpoint | Description |
+|---|---|---|
+| GET | `/` | Service root — model status & info |
+| GET | `/health` | Health check — service & model availability |
+| GET | `/metrics` | Prometheus metrics endpoint |
+| GET | `/models/status` | Detailed model load status |
+| GET | `/api/ab-status` | A/B testing configuration status |
+| POST | `/api/upload` | Upload an image, returns `image_id` |
+| POST | `/api/classify` | Classify by `image_id` |
+| GET | `/api/classify/{image_id}` | Get stored classification result |
+| POST | `/api/segment` | Segment by `image_id` |
+| GET | `/api/segment/{image_id}` | Get stored segmentation result |
+| GET | `/api/images/{image_id}` | Retrieve uploaded image |
+
+## Environment Variables
+
+| Variable | Default | Description |
+|---|---|---|
+| `HF_TOKEN` | - | Hugging Face token for model downloads |
+| `HF_MODEL_REPO` | `omarelrayes/mlflow-artifacts` | HF repo with model zips |
+| `CLASSIFIER_MODEL_PATH` | `models/classifier_savedmodel.zip` | Path in HF repo |
+| `SEGMENTER_MODEL_PATH` | `models/segmenter_savedmodel.zip` | Path in HF repo |
+| `CLASSIFIER_THRESHOLD` | `0.5` | Confidence threshold |
+| `AB_TESTING_ENABLED` | `false` | Enable A/B testing |
+| `AB_CLASSIFIER_B` | - | Alternative model URI |

requirements.txt

@@ -1,8 +1,10 @@
-fastapi
-uvicorn
-tensorflow>=2.18.0
-mlflow
-huggingface_hub
-python-multipart
-pillow
-numpy
+fastapi>=0.104.0
+uvicorn[standard]>=0.24.0
+python-multipart>=0.0.6
+Pillow>=10.1.0
+numpy>=1.24.3
+pydantic>=2.5.0
+tensorflow-cpu>=2.13.0
+prometheus-client>=0.19.0
+huggingface-hub>=0.20.0
+starlette>=0.27.0