adwod/Streamlite_ViT_2000

1import streamlit as st2from transformers import pipeline3from PIL import Image4from transformers import (5    AutoImageProcessor,6    AutoModelForImageClassification7)8 9pre_process = AutoImageProcessor.from_pretrained("adwod/ViT_bluetab")10 11 12st.title("What is it?")13 14file_name = st.file_uploader("Upload a hot dog candidate image")15 16if file_name is not None:17    col1, col2 = st.columns(2)18 19    image = Image.open(file_name)20    col1.image(image, use_column_width=True)21    inputs = pre_process(images=image, return_tensors="pt")22    input_pixels = inputs.pixel_values23    model = AutoModelForImageClassification.from_pretrained("adwod/ViT_bluetab")24    outputs = model(input_pixels)25    col2.header(model.config.id2label[outputs.logits.argmax(-1)  .item()])