from transformers import pipeline
import numpy
import torch
import matplotlib.pyplot
import PIL
from PIL import Image
import streamlit as st

st.write('age detection')
upload_file = st.file_uploader('choose a image file',type='jpg')

pipe = pipeline("image-classifcation",model='dima806/facial_age_image_detection')

st.write(pipe(st.image(Image.open(upload_file)))[0]['label'])