doctoria-rally-ai 🩺
Fine-tuned medical Vision-Language Model for SahhaAI — offline, private wound-care & medical VQA for disconnected clinics in Morocco.
- Base:
HuggingFaceTB/SmolVLM-256M-Instruct(SmolVLM) - Method: LoRA fine-tune (PyTorch + PEFT, trained locally on Apple Silicon / MPS)
- Data: VQA-RAD (radiology VQA, CC0)
- Runs: in-browser (WebGPU via transformers.js) and locally — data stays on device
Benchmark (VQA-RAD test)
- Closed-ended accuracy: 0.439
- Open-ended token-F1: 0.228
- Speed: 19.0 tok/s on Apple Silicon (MPS)
- See repo
finetune/BENCHMARK.mdfor base-vs-fine-tuned analysis.
Use
from transformers import AutoProcessor, AutoModelForImageTextToText
from PIL import Image
m = AutoModelForImageTextToText.from_pretrained("doctoria/doctoria-rally-ai")
p = AutoProcessor.from_pretrained("doctoria/doctoria-rally-ai")
msgs = [{"role":"user","content":[{"type":"image"},{"type":"text","text":"Assess this wound."}]}]
text = p.apply_chat_template(msgs, add_generation_prompt=True)
inp = p(text=text, images=[[Image.open("wound.jpg")]], return_tensors="pt")
print(p.batch_decode(m.generate(**inp, max_new_tokens=128), skip_special_tokens=True)[0])
⚠ Decision support, not a diagnosis. A trained health worker stays in the loop.
— Author: Jad Tounsi El Azzouzi · part of SahhaAI · Apache-2.0