import tensorflow as tf from transformers import pipeline import streamlit as st model = pipeline('image-classification', model='google/vit-base-patch16-224-in21k') result = model('path/to/image.jpg') print(result)