age_detection / app.py
corvo7's picture
Update app.py
013cf16 verified
raw
history blame
394 Bytes
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")
pipe = pipeline("image-classification", model="dima806/facial_age_image_detection")
st.write(pipe(st.image(Image.open(uploaded_files )))[0]["label"])