Views
No views yet
1from transformers import AutoModelForImageClassification, AutoProcessor
2
3model = AutoModelForImageClassification.from_pretrained("ozair23/autotrain-w5nk2-rvmqx")
4processor = AutoProcessor.from_pretrained("ozair23/autotrain-w5nk2-rvmqx")
5
6def predict(image):
7 inputs = processor(images=image, return_tensors="pt")
8 outputs = model(**inputs)
9 return outputs