README.md

---
title: ResNet34 Corrosion Classifier
emoji: 🧪
colorFrom: indigo
colorTo: green
sdk: gradio
sdk_version: "4.44.1"
app_file: app.py
pinned: false
---

# ResNet34 Corrosion Classifier — Hugging Face Space

Semplice Space Gradio che carica un modello ResNet34 e predice 9 classi di corrosione.

## File
- `app.py`: interfaccia Gradio.
- `model.py`: definizione modello e caricamento pesi.
- `classes.json`: etichette delle classi.
- `requirements.txt`: dipendenze.
- `.gitattributes`: abilita LFS per i file `.pth`.
- `resnet34_best.pth`: **DA CARICARE DA TE** (non incluso).

## Istruzioni
1. Crea una nuova Space su Hugging Face (Gradio + Python).
2. Carica questi file nella Space.
3. Aggiungi il tuo file di pesi `resnet34_best.pth` (usa Git LFS se > 50 MB).
4. (Opzionale) Se il file si chiama diversamente, imposta la variabile d'ambiente `CKPT_PATH`
   nelle Settings della Space, oppure modifica `CKPT_PATH` in `app.py`.
5. Avvia la Space.

## Uso
- Carica o scatta una foto, poi clicca **Analizza immagine**.
- La card a destra mostra le probabilità (Top-K) e la predizione.

## Note
- Il modello gira su CPU per default. Se vuoi più velocità, passa a una Space con GPU.
- Le trasformazioni input usano Resize 256 → CenterCrop 224 e normalizzazione ImageNet.