0xgr3y commited on
Commit
572f5f8
·
verified ·
1 Parent(s): 7074b81

Fix citation title, add download instructions for TF-Lite/SavedModel snippets

Browse files
Files changed (1) hide show
  1. README.md +18 -1
README.md CHANGED
@@ -388,6 +388,13 @@ print(f"Predicted: {LABELS[np.argmax(preds)]} ({np.max(preds)*100:.1f}%)")
388
 
389
  ### Python — TF-Lite
390
 
 
 
 
 
 
 
 
391
  ```python
392
  import numpy as np
393
  import tensorflow as tf
@@ -416,6 +423,14 @@ for i in top3_idx:
416
 
417
  ### Python — SavedModel
418
 
 
 
 
 
 
 
 
 
419
  ```python
420
  import tensorflow as tf
421
  import numpy as np
@@ -488,7 +503,9 @@ print(f"Predicted: {LABELS[np.argmax(preds)]} ({np.max(preds)*100:.1f}%)")
488
 
489
  ```bibtex
490
  @misc{saugani2026_arch_building,
491
- title={Architecture Building Image Classifier: FGVC with DenseNet121 + GeM Pooling + SWA},
 
 
492
  author={Saugani},
493
  year={2026},
494
  publisher={Hugging Face},
 
388
 
389
  ### Python — TF-Lite
390
 
391
+ Download the model first:
392
+ ```python
393
+ from huggingface_hub import hf_hub_download
394
+ hf_hub_download("0xgr3y/Arch-Building-Image-Classification", "tflite/model.tflite", local_dir=".")
395
+ hf_hub_download("0xgr3y/Arch-Building-Image-Classification", "tflite/label.txt", local_dir=".")
396
+ ```
397
+
398
  ```python
399
  import numpy as np
400
  import tensorflow as tf
 
423
 
424
  ### Python — SavedModel
425
 
426
+ Download the model first:
427
+ ```python
428
+ from huggingface_hub import snapshot_download
429
+ snapshot_download("0xgr3y/Arch-Building-Image-Classification", allow_patterns=["saved_model/*"], local_dir=".")
430
+ ```
431
+
432
+ Requires the custom layer definitions from the Keras section above.
433
+
434
  ```python
435
  import tensorflow as tf
436
  import numpy as np
 
503
 
504
  ```bibtex
505
  @misc{saugani2026_arch_building,
506
+ title={Fine-Grained Visual Categorization of World Architectural Buildings
507
+ Using CNN Transfer Learning DenseNet121 with Fine-Tuning and
508
+ Multi-Layer Regularization Strategy},
509
  author={Saugani},
510
  year={2026},
511
  publisher={Hugging Face},