Views
No views yet
1from transformers import AutoProcessor
2from src.models.qwen_classifier import Qwen3VLClassifier
3import torch
4from PIL import Image
5
6# Load model and processor
7model = Qwen3VLClassifier.from_pretrained("your-username/model-name")
8processor = AutoProcessor.from_pretrained("oscarqjh/Qwen3-VL-4B-Instruct-Flowers102-Open-QA")
9
10# Prepare inputs
11image = Image.open("flower.jpg")
12text = "What type of flower is this?"
13
14# Create chat format
15messages = [{
16 'role': 'user',
17 'content': [
18 {'type': 'image', 'image': image},
19 {'type': 'text', 'text': text}
20 ]
21}]
22
23# Process inputs
24text_input = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=False)
25inputs = processor(text=[text_input], images=[image], return_tensors="pt", padding=True)
26
27# Get predictions
28with torch.no_grad():
29 outputs = model(**inputs)
30 predicted_class = outputs.logits.argmax(dim=-1).item()
31 confidence = torch.softmax(outputs.logits, dim=-1).max().item()
32
33print(f"Predicted class: {predicted_class}, Confidence: {confidence:.4f}")