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