Hugging Face's logo

Spaces:
habeebCycle
/
Beit-Retinal
Sleeping

App Files Community
Beit-Retinal
File size: 774 Bytes 
7113b7f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
---
title: Beit-Retinal
emoji: 🚀
colorFrom: red
colorTo: green
sdk: docker
pinned: false
license: mit
short_description: Training model for some retina dataset
---

# RETFound MAE – Hugging Face Space (FastAPI): Train + Inference

This Space lets you upload a zipped ImageFolder dataset, fine-tune a classifier head on top of RETFound MAE, run predictions, and push the trained model to the Hub.

## Endpoints
- `POST /predict` → form-data file=`image.jpg`

## Env Vars
- `HF_BASE_MODEL_REPO` (e.g., `username/retfound-model`)

## Dataset format
```

zip-root/
train/
ClassA/*.jpg
ClassB/*.jpg
val/
ClassA/*.jpg
ClassB/*.jpg

```

## Notes
- GPU recommended. If CPU-only, reduce batch size.
- Validation accuracy is reported; best checkpoint saved to `checkpoints/`.