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']) | |