app.py

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(st.image(uploaded_files))[0]["label"])