feat(Admin > Walk): Display aesthetic score per thumb

Author: danactive

Makefile

@@ -31,9 +31,9 @@ build-ai-api:
 	docker build -f apps/api/Dockerfile -t ai-api .
 
 ai-api:
-	docker run --name ai-api-run -p 8080:8080 ai-api > api.log 2>&1 || true
-	cat api.log
-	docker rm ai-api-run || true
+	# OpenAI model stores in ~/.cache/clip
+	docker run --rm --name ai-api -p 8080:8080 \
+		-v $(HOME)/.cache/clip:/root/.cache/clip
 
 build-test:
 	docker build  -f apps/api/Dockerfile --build-arg INSTALL_TEST=true -t ai-api-test .

apps/api/README.md

@@ -29,13 +29,21 @@ This project serves a high-accuracy image classification API using a **Vision Tr
 
 # Aesthetic Scoring
 
-## Model details
-
-- **[OpenCLIP](https://github.com/mlfoundations/open_clip) ViT-L/14** model (CLIP architecture)
-- **Checkpoint** laion2B-s32B-b82K
-- **Aesthetic scoring** using a trained multilayer perceptron (MLP)
-- Deterministic evaluation
-- A custom regression MLP head trained on:
-  	- [AVA dataset (aesthetics annotated)](https://github.com/microsoft/LMOps/blob/main/promptist/aesthetic/ava%2Blogos-l14-linearMSE.pth)
-	- Logos subset (for composition)
-- [Code sample](https://github.com/christophschuhmann/improved-aesthetic-predictor/tree/main)
+This project provides an API for **aesthetic scoring** of images using a regression head trained on top of OpenAI's CLIP ViT-B/16 backbone.
+
+---
+
+## 🧠 Model Details
+
+- **Backbone:** OpenAI CLIP ViT-B/16
+- **Regression Head:** Multilayer Perceptron (MLP) trained for aesthetic prediction
+- **Head Weights:** `models/aesthetic/sa_0_4_vit_b_16_linear.pth`
+- **Feature Dimension:** 512
+
+
+## Local setup
+
+1. Download the regression head:
+   [sa_0_4_vit_b_16_linear.pth](https://github.com/LAION-AI/aesthetic-predictor/blob/main/sa_0_4_vit_b_16_linear.pth)
+1. Place it in `models/aesthetic/sa_0_4_vit_b_16_linear.pth`
+1. The OpenAI CLIP backbone weights for ViT-B/16 will be downloaded automatically on first run `make ai-api`

apps/api/main.py

@@ -9,7 +9,7 @@
 # Setup logging once
 logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
 logger = logging.getLogger("uvicorn")
-logger.setLevel(logging.ERROR)
+logger.setLevel(logging.DEBUG)
 
 logger.debug("FastAPI is initializing...")
 

apps/load-weights/README.md

@@ -10,6 +10,4 @@ Hugging Face timm https://huggingface.co/timm/convnext_large_mlp.laion2b_ft_augr
 
 ## Aesthetic Scorer
 
-`make load-weights MODEL_REPO=LAION/CLIP-based-Aesthetic-Predictor FILENAMES="ava+logos-l14-linearMSE.pth"`
-
-https://github.com/LAION-AI/aesthetic-predictor/blob/main/sa_0_4_vit_b_16_linear.pth
+No longer uses weights loader