import streamlit as st
from transformers import pipeline
import numpy
import torch

st.write("welcome to age detection app")

uploaded_files = st.file_uploader("Choose a image file")

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

st.write(pipe(uploaded_files)[0]["label"])