Push model using huggingface_hub.

README.md

@@ -3,73 +3,8 @@ pipeline_tag: image-classification
 tags:
 - model_hub_mixin
 - pytorch_model_hub_mixin
-- OCT
-- retinal-imaging
-- classification
 ---
 
-## RetFiner-UrFound
-This repo contains the weights of RetFiner-UrFound 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 this 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_base_patch16
-
-
-class RetFiner(nn.Module, PyTorchModelHubMixin):
-    """RetFiner: Fine-tuned ViT models for retinal image analysis"""
-    
-    def __init__(self, model_name: str = "RetFiner-UrFound", num_classes: int = 2, **kwargs):
-        super().__init__()
-        
-        self.model = vit_base_patch16(
-            img_size=224,
-            num_classes=num_classes,
-            drop_path_rate=0.1,
-            global_pool=False,
-            use_proj=False,
-            **kwargs
-        )
- 
-        self.config = {
-            "num_classes": num_classes,
-            **kwargs
-        }
-
-    def forward(self, x):
-        return self.model(x)
-
-model = RetFiner.from_pretrained('ronnief1/RetFiner-UrFound')
-```
-
-## 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:c96ebb8751171760197e19a4896f23ceb53a78113b9bae8cb36d64f7f5d98bf7
-size 343215776
+oid sha256:9727f3662ef3fd159888882fb7078582695dc552ee4aeb40f84105d412fd44e2
+size 343209448