app.py

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



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

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

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

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