Create app.py

app.py

@@ -0,0 +1,31 @@
+import streamlit as st
+from transformers import pipeline
+from PIL import Image
+
+# Load the classifier model
+@st.cache_resource
+def load_classifier():
+    return pipeline("image-classification", model="nateraw/vit-age-classifier")
+
+classifier = load_classifier()
+
+# Streamlit UI
+st.title("Age Classifier App 🧑👵")
+st.write("Upload an image to predict the age category.")
+
+# Upload an image
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
+
+if uploaded_file is not None:
+    # Load and display the image
+    image = Image.open(uploaded_file).convert("RGB")
+    st.image(image, caption="Uploaded Image", use_column_width=True)
+
+    # Classify the age
+    st.write("Classifying...")
+    results = classifier(image)
+
+    # Display results
+    st.subheader("Results:")
+    for result in results:
+        st.write(f"**{result['label']}** - Score: {result['score']:.4f}")