Spaces:
Sleeping
Sleeping
File size: 1,279 Bytes
aa983ba 9c9a989 aa983ba d733d16 aa983ba |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
---
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.
|