OwensLab/CommunityForensics-Small
Viewer • Updated • 356k • 2.1k
How to use OwensLab/commfor-data-preprocessor with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("OwensLab/commfor-data-preprocessor", dtype="auto")This is a data processor for the Community Forensics classifiers. Detailed instructions will be updated at the Community Forensics GitHub_repo in the future.
dataprocessor_hf.py is included in this repository.
import dataprocessor_hf as dphf
from PIL import Image
data_processor = dphf.CommForImageProcessor.from_pretrained('OwensLab/commfor-data-preprocessor', size=384)
img = Image.open('path_to_img.png')
processed_image = data_processor(img, mode='test')
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("OwensLab/commfor-data-preprocessor", dtype="auto")