File size: 743 Bytes
f593fb6 a2105df f593fb6 a2105df f593fb6 a2105df f593fb6 d20ce09 a2105df | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | import streamlit as st
from transformers import pipeline
from PIL import Image
from transformers import (
AutoImageProcessor,
AutoModelForImageClassification
)
pre_process = AutoImageProcessor.from_pretrained('adwod/VIT_10000')
st.title("What is it?")
file_name = st.file_uploader("Upload candidate image")
if file_name is not None:
col1, col2 = st.columns(2)
image = Image.open(file_name)
col1.image(image, use_column_width=True)
inputs = pre_process(images=image, return_tensors="pt")
input_pixels = inputs.pixel_values
model = AutoModelForImageClassification.from_pretrained('adwod/VIT_10000')
outputs = model(input_pixels)
col2.header(model.config.id2label[outputs.logits.argmax(-1) .item()]) |