import streamlit as st
from transformers import pipeline
from PIL import Image
import requests
pip install transformers torch


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 ของรูปภาพ", "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("ส่งคำขอ")
        
        if clicked:
            results = classifier(img)
            st.json(results)

if __name__ == "__main__":
    main()