README.md

metadata

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.