|
import streamlit as st |
|
from transformers import pipeline |
|
from PIL import Image |
|
import requests |
|
|
|
classifier = pipeline("image-classification", model="google/vit-base-patch16-224") |
|
def get_img_from_url(url): |
|
return Image.open(requests.get(url, stream=True).raw) |
|
|
|
def main(): |
|
st.title("image-classification") |
|
|
|
with st.form("image"): |
|
|
|
url = st.text_input("URL to some image", "https://images.livemint.com/img/2022/08/01/600x338/Cat-andriyko-podilnyk-RCfi7vgJjUY-unsplash_1659328989095_1659328998370_1659328998370.jpg") |
|
img = get_img_from_url(url) |
|
|
|
clicked = st.form_submit_button("Submit") |
|
if clicked: |
|
results = classifier(img) |
|
st.json(results) |
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
main() |
