Push model using huggingface_hub.

README.md

@@ -3,75 +3,8 @@ pipeline_tag: image-classification
 tags:
 - model_hub_mixin
 - pytorch_model_hub_mixin
-- OCT
-- retinal-imaging
-- classification
 ---
 
-
-## RetFiner-RETFound
-This repo contains the weights of RetFiner-RETFound from the paper [RetFiner: A Vision-Language Refinement Scheme for Retinal Foundation Models](https://arxiv.org/abs/2506.22149).
-
-Project page: [RetFiner](https://github.com/ronnief1/RetFiner)
-
-
-### Required Libraries
-This model requires specific Python libraries:
-```bash
-torch==2.4.1+cu118
-timm==0.4.12
-torchvision==0.19.1+cu118
-```
-
-To use the model, please download the ViT implementation from [vit.py](https://github.com/ronnief1/RetFiner/blob/main/RetFiner/vit.py)
-
-Note: if you are using this for downstream inference, adjust the num_classes based on the target dataset.
-
-```python
-import torch
-import torch.nn as nn
-from huggingface_hub import PyTorchModelHubMixin
-from vit import vit_large_patch16
-
-class RetFiner(nn.Module, PyTorchModelHubMixin):
-    """RetFiner: Fine-tuned ViT models for retinal image analysis"""
-    
-    def __init__(self, model_name: str = "RetFiner-RETFound", num_classes: int = 2, **kwargs):
-        super().__init__()
-        
-        self.model = vit_large_patch16(
-                img_size=224,
-                num_classes=num_classes,
-                drop_path_rate=0.1,
-                global_pool=False,
-                use_proj=True,
-                **kwargs
-            )
-        
-        self.config = {
-            "num_classes": num_classes,
-            **kwargs
-        }
-
-    def forward(self, x):
-        return self.model(x)
-
-model = RetFiner.from_pretrained('ronnief1/RetFiner-RETFound')
-```
-
-
-## Citation
-
-Please cite the original paper if you use this model:
-
-```python
-@misc{fecso2025retfinervisionlanguagerefinementscheme,
-      title={RetFiner: A Vision-Language Refinement Scheme for Retinal Foundation Models}, 
-      author={Ronald Fecso and José Morano and Ursula Schmidt-Erfurth and Hrvoje Bogunović},
-      year={2025},
-      eprint={2506.22149},
-      archivePrefix={arXiv},
-      primaryClass={cs.CV},
-      url={https://arxiv.org/abs/2506.22149}, 
-}
-```
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Library: ronnief1/RetFiner
+- Docs: [More Information Needed]

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ca1aca0c2dcac3382b9b96c21049915ac6f24c09ad48639332f5e36104b9b857
-size 1216393632
+oid sha256:72c72a88f4e9cb81f1cfba56bbdeeccfeee36e0b741fce8748e0b4d0020174c3
+size 1216385256