corvo7 commited on
Commit
94a4f55
·
verified ·
1 Parent(s): 3f536ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline
3
+ import numpy
4
+ import torch
5
+
6
+ st.write("welcome to age detection app")
7
+
8
+ uploaded_files = st.file_uploader("Choose a image file")
9
+
10
+ pipe = pipeline("image-classification", model="dima806/facial_age_image_detection")
11
+
12
+ st.write(pipe(uploaded_files)[0]["label"])