Update model card

README.md

@@ -5,57 +5,26 @@ tags:
   - vision-transformer
   - civitai
   - vit5
+  - runpod
+pipeline_tag: image-classification
 ---
 
 # GenGlasses — CivitAI Model Reverse Engineer
 
-Predicts the **Stable Diffusion model**, **model family**, and **tags**
-from a generated image, using a fine-tuned [ViT-5](https://arxiv.org/abs/2602.08071) backbone.
+Predicts **SD checkpoint**, **model family**, and **tags** from a generated image.  
+Served via RunPod serverless: [cjyu81/genglassrunpod](https://github.com/cjyu81/genglassrunpod)
 
 ## Label Space
-- **Models** : 50 (top-50 CivitAI checkpoints by downloads)
+- **Models**  : 50
 - **Families**: 8 — `Flux, Illustrious, NoobAI, Other, Pony, SD 1.5, SDXL, ZImageTurbo`
 - **Tags**    : 108
 
-## Architecture
-- Backbone : `vit5_base` (pretrained weights from FengWang3211/ViT-5)
-- 3 output heads: `head_model` (CE), `head_family` (CE), `head_tags` (BCE multi-label)
-
 ## Training History
-| Run | Date | New Images | Best Val Loss |
-|-----|------|-----------|---------------|
-| 1 | 2026-04-04 | 834 | 2.3399 |
-
-
-## Usage
-```python
-import torch, json
-from huggingface_hub import hf_hub_download
-from PIL import Image
-from torchvision import transforms
-
-ckpt_path  = hf_hub_download('Charlie81/genglasses', 'best.pth')
-maps_path  = hf_hub_download('Charlie81/genglasses', 'label_maps.json')
-
-with open(maps_path) as f:
-    label_maps = json.load(f)
-
-ckpt = torch.load(ckpt_path, map_location='cpu')
-# Re-build model (see training notebook) then:
-model.load_state_dict(ckpt['model'])
-model.eval()
-
-img = Image.open('your_image.jpg').convert('RGB')
-tf  = transforms.Compose([
-    transforms.Resize(256), transforms.CenterCrop(224),
-    transforms.ToTensor(),
-    transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225]),
-])
-with torch.no_grad():
-    out_model, out_family, out_tags = model(tf(img).unsqueeze(0))
+| Run | Date | Skip Scrape | Fresh Train | New Images | Best Val Loss |
+|-----|------|------------|-------------|-----------|---------------|
+| 1 | 2026-04-04 | True | True | 831 | 2.3535 |
 
-pred_model  = label_maps['model_names'][out_model.argmax().item()]
-pred_family = label_maps['model_families'][out_family.argmax().item()]
-pred_tags   = [label_maps['all_tags'][i] for i, v in enumerate(out_tags.sigmoid()[0]) if v > 0.5]
-print(pred_model, pred_family, pred_tags)
+## RunPod Request
+```json
+{"input": {"image": "<base64-jpeg>"}}
 ```