Downloads · 30 days
1.1K
14% of all-time downloads
NYUAD-ComNets/FaceScanPaliGemma_Race
FaceScanPaliGemma_Race is a machine learning model from NYUAD-ComNets. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as gemma.
should probably proofread and complete it, then remove this comment. --
Downloads · 30 days
1.1K
14% of all-time downloads
All-time downloads
7.3K
Public
Parameters
2.9B
5.8 GB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors5.8 GB · 100%
From the Hugging Face model README
from PIL import Image
import torch
from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/FaceScanPaliGemma_Race',torch_dtype=torch.bfloat16)
input_text = "what is the race of the person in the image?"
processor = PaliGemmaProcessor.from_pretrained("google/paligemma-3b-pt-224")
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)
input_image = Image.open('image_path')
inputs = processor(text=input_text, images=input_image, padding="longest", do_convert_rgb=True, return_tensors="pt").to(device)
inputs = inputs.to(dtype=model.dtype)
with torch.no_grad():
output = model.generate(**inputs, max_length=500)
result=processor.decode(output[0], skip_special_tokens=True)[len(input_text):].strip()
from transformers import AutoProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig
from PIL import Image
import requests
import torch
import time
device = "cuda:0"
dtype = torch.bfloat16
quantization_config = BitsAndBytesConfig(load_in_8bit=True)
model = PaliGemmaForConditionalGeneration.from_pretrained(
"NYUAD-ComNets/FaceScanPaliGemma_Race", quantization_config=quantization_config
).eval()
processor = AutoProcessor.from_pretrained("google/paligemma-3b-pt-224")
prompt = "what is the race of the person in the image?"
image = Image.open('image_path')
model_inputs = processor(text=prompt, images=image, return_tensors="pt").to(model.device)
input_len = model_inputs["input_ids"].shape[-1]
with torch.inference_mode():
generation = model.generate(**model_inputs, max_new_tokens=100, do_sample=False)
generation = generation[0][input_len:]
decoded = processor.decode(generation, skip_special_tokens=True)
print(decoded)
This model is a fine-tuned version of google/paligemma-3b-pt-224 on the FairFace dataset. The model aims to classify the race of face image or image with one person into seven categoris such as Black, East Asian, Indian, Latino_Hispanic, Middle Eastern, Southeast Asian, White
Model Performance
Accuracy: 81 %, F1 score: 79 %
This model is used for research purposes
FairFace dataset was used for training and validating the model
The following hyperparameters were used during training:
@article{aldahoul2026facescanpaligemma,
title={FaceScanPaliGemma multi-agent vision language models for facial attribute recognition},
author={AlDahoul, Nouar and Tan, Myles Joshua Toledo and Kasireddy, Harishwar Reddy and Zaki, Yasir},
journal={Scientific Reports},
year={2026},
publisher={Nature Publishing Group UK London}
}
@article{aldahoul2024exploring,
title={Exploring Vision Language Models for Facial Attribute Recognition: Emotion, Race, Gender, and Age},
author={AlDahoul, Nouar and Tan, Myles Joshua Toledo and Kasireddy, Harishwar Reddy and Zaki, Yasir},
journal={arXiv preprint arXiv:2410.24148},
year={2024}
}
@misc{ComNets,
url={https://huggingface.co/NYUAD-ComNets/FaceScanPaliGemma_Race](https://huggingface.co/NYUAD-ComNets/FaceScanPaliGemma_Race)},
title={FaceScanPaliGemma_Race},
author={Nouar AlDahoul, Yasir Zaki}
}
The FaceScanPaliGemma model processes highly sensitive biometric data (facial attributes). Deployment of this model must follow strict governance frameworks to ensure responsible and ethical use.
We encourage the community to report issues, biases, or misuse of this model through the Hugging Face Hub discussion forum.