Upload README.md with huggingface_hub

README.md

@@ -1,6 +1,7 @@
 ---
 library_name: timm
 pipeline_tag: image-classification
+base_model: google/efficientnet-b0
 tags:
   - anime-classification
   - real-photos
@@ -15,11 +16,14 @@ inference: true
 
 # Anime/Real/Rendered Image Classifier (EfficientNet-B0)
 
-**Fast, lightweight classifier for distinguishing photographs from anime and 3D rendered images.**
+Fine-tuned EfficientNet-B0 for distinguishing photographs from anime and 3D rendered images.
+
+**Base Model:** [google/efficientnet-b0](https://huggingface.co/google/efficientnet-b0)
 
 ## Model Details
 
 - **Architecture:** EfficientNet-B0 (timm)
+- **Base Model:** Google's EfficientNet-B0 (ImageNet pretrained)
 - **Input Size:** 224×224 RGB
 - **Classes:** anime, real, rendered
 - **Parameters:** 5.3M
@@ -80,6 +84,7 @@ print(f"{labels[pred_class]}: {probs[0, pred_class]:.2%}")
 
 ## Training Details
 
+- **Base Model:** ImageNet-pretrained EfficientNet-B0
 - **Augmentation:** Raw (resize only)
 - **Optimizer:** AdamW (lr=0.001)
 - **Loss:** CrossEntropyLoss with class weighting