Upload 5 files

README.md

@@ -1,8 +1,37 @@
+---
+title: Corrosion Classifier (ViT-B/16 • CPU)
+emoji: 🧪
+colorFrom: indigo
+colorTo: green
+sdk: gradio
+sdk_version: "4.44.0"
+app_file: app.py
+pinned: false
+---
+
 # Corrosion Classifier (ViT-B/16 · CPU)
-Space Gradio pronto per Zero GPU. Carica `vit_b16_best.pth` nella root.
 
-## Uso
-1. Crea uno Space Gradio su Hugging Face.
-2. Carica: app.py, requirements.txt, classes.json, vit_b16_best.pth.
-3. Runtime: CPU.
-4. Avvia.
+Space Gradio pronto per **Zero GPU**. Carica `vit_b16_best.pth` nella root dello Space e gira su CPU.
+
+## Come usare
+1. Crea uno Space **Gradio** su Hugging Face.
+2. Carica questi file nella root dello Space:
+   - `app.py`
+   - `requirements.txt`
+   - `classes.json`
+   - **`vit_b16_best.pth`** (i pesi del tuo modello)
+   - opz.: `.gitattributes`
+3. In **Settings → Runtime → Hardware** seleziona **CPU**.
+4. Avvia lo Space.
+
+## Variabili opzionali
+- `MODEL_WEIGHTS_PATH`: percorso alternativo ai pesi (default: `vit_b16_best.pth`).
+- `CLASSES_PATH`: percorso per la lista classi (default: `classes.json`).
+
+## Dettagli tecnici
+- Modello: `timm.create_model("vit_base_patch16_224", num_classes=9)`.
+- Transform: resize 224 + normalizzazione ImageNet.
+- Output: etichetta principale + **Top-3** con percentuali.
+
+## Debug
+Se compaiono "missing/unexpected keys" in `load_state_dict`, è normale quando i pesi hanno prefissi (`module.`/`model.`) o layer EMA. Il caricamento usa `strict=False` per tollerarle.