b932af1 514ab64 b932af1
1
2
3
4
5
6
7
8
9
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)