import torch
from transformers import AutoModel, AutoConfig
from torchvision import transforms
from PIL import Image
import os

os.chdir(os.path.dirname(os.path.abspath(__file__)))

# Load model and config
model = AutoModel.from_pretrained(
    "./classifier-mix",  # or path to your model directory
    trust_remote_code=True
)
model.eval()

# Load and preprocess image
img = Image.open("classifier-mix/sample_img.png").convert("RGB")

outputs = model(img)

print("Predicted class:", outputs[0]['label'])
print("Confidences:", outputs[0]['confidences'])