| from transformers import pipeline | |
| from PIL import Image | |
| import streamlit as st | |
| classifier = pipeline("image-classification", model="research/ai_detector_v2/checkpoint-1815") | |
| with st.sidebar: | |
| image= st.sidebar.file_uploader("Input image:", type=["png", "jpg"]) | |
| btn= st.button("Submit") | |
| if btn: | |
| img= original_image= Image.open(image) | |
| st.json( | |
| classifier(img) | |
| ) |